diff --git a/hummingbird/ml/operator_converters/onnx/array_feature_extractor.py b/hummingbird/ml/operator_converters/onnx/array_feature_extractor.py index 4b49acc33..fc29304a7 100644 --- a/hummingbird/ml/operator_converters/onnx/array_feature_extractor.py +++ b/hummingbird/ml/operator_converters/onnx/array_feature_extractor.py @@ -45,3 +45,4 @@ def convert_onnx_array_feature_extractor(operator, device, extra_config): register_converter("ONNXMLArrayFeatureExtractor", convert_onnx_array_feature_extractor) +register_converter("ONNXMLGather", convert_onnx_array_feature_extractor) diff --git a/hummingbird/ml/supported.py b/hummingbird/ml/supported.py index 14ea8574a..e4f1ff52a 100644 --- a/hummingbird/ml/supported.py +++ b/hummingbird/ml/supported.py @@ -89,6 +89,7 @@ Concat, Div, Imputer, +Gather, LabelEncoder, Less, LinearClassifier, @@ -357,6 +358,7 @@ def _build_onnxml_operator_list(): "ArrayFeatureExtractor", "Binarizer", "FeatureVectorizer", + "Gather", "Imputer", "LabelEncoder", "OneHotEncoder",