diff options
author | Lester Solbakken <lesters@oath.com> | 2020-04-29 13:54:39 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2020-04-29 13:54:39 +0200 |
commit | e5abba1bd7ebd7887246190914a0c7501ec019d1 (patch) | |
tree | 515f1f401a279b301a272bba626dc9c51abaf189 /model-integration | |
parent | a8fc679943ae9bdc6d6d58d82a2b9f43dac54ad4 (diff) |
Set default tensor type to float for onnx models
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"); } |