diff options
author | Lester Solbakken <lesters@oath.com> | 2021-06-08 09:44:31 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2021-06-08 09:44:31 +0200 |
commit | 80dde1d8e61a9ff87b3c5a1a5ce8db267a587eb8 (patch) | |
tree | b4ae58235937813c80e8e68b620d806bad32b51c /model-integration | |
parent | d4c2da810633ed21a265e634e951c779fccb4339 (diff) |
Upgrade Java ONNX Runtime to 1.8
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/test/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorTest.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model-integration/src/test/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorTest.java b/model-integration/src/test/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorTest.java index 4b42e18d75e..a7186aae5fe 100644 --- a/model-integration/src/test/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorTest.java +++ b/model-integration/src/test/java/ai/vespa/modelintegration/evaluator/OnnxEvaluatorTest.java @@ -74,7 +74,7 @@ public class OnnxEvaluatorTest { assertEvaluate("cast_int8_float.onnx", "tensor<float>(d0[1]):[-128]", "tensor<int8>(d0[1]):[128]"); assertEvaluate("cast_float_int8.onnx", "tensor<int8>(d0[1]):[-1]", "tensor<float>(d0[1]):[255]"); - // ONNX Runtime 1.7.0 does not support much of bfloat16 yet + // ONNX Runtime 1.8.0 does not support much of bfloat16 yet // assertEvaluate("cast_bfloat16_float.onnx", "tensor<float>(d0[1]):[1]", "tensor<bfloat16>(d0[1]):[1]"); } |