microsoft / hummingbird

Hummingbird compiles trained ML models into tensor computation for faster inference.
MIT License
3.32k stars 274 forks source link

Adding more ONNX data types #743

Open jjasont opened 7 months ago

jjasont commented 7 months ago

The current Hummingbird implementation only convert 5 types of variable to the relevant ONNX type in here, namely:

We can consider adding conversion to BooleanTensorType and DictionaryType to account the corresponding boolean and dictionary input.