diff options
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TypeConverter.java | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TypeConverter.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TypeConverter.java index 9354a346aaf..ef7038b1793 100644 --- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TypeConverter.java +++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TypeConverter.java @@ -64,12 +64,12 @@ class TypeConverter { case BOOL: return TensorType.Value.FLOAT; case INT8: return TensorType.Value.FLOAT; case INT16: return TensorType.Value.FLOAT; - case INT32: return TensorType.Value.DOUBLE; - case INT64: return TensorType.Value.DOUBLE; + case INT32: return TensorType.Value.FLOAT; + case INT64: return TensorType.Value.FLOAT; case UINT8: return TensorType.Value.FLOAT; case UINT16: return TensorType.Value.FLOAT; - case UINT32: return TensorType.Value.DOUBLE; - case UINT64: return TensorType.Value.DOUBLE; + case UINT32: return TensorType.Value.FLOAT; + case UINT64: return TensorType.Value.FLOAT; default: throw new IllegalArgumentException("A ONNX tensor with data type " + dataType + " cannot be converted to a Vespa tensor type"); } |