diff options
author | Lester Solbakken <lesters@oath.com> | 2021-05-21 09:33:32 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2021-05-21 09:33:32 +0200 |
commit | 6448742f804482946a7bf2d17723dca6b4100b73 (patch) | |
tree | 135038f0298f3e519ed8e4327cf1bf1915df4b39 /config-model/src/test/cfg/application/ml_serving/models/sqrt.py | |
parent | 864eb3da782e9795826ec78add953a76eeb2ea17 (diff) |
Wire in stateless ONNX runtime evaluation
Diffstat (limited to 'config-model/src/test/cfg/application/ml_serving/models/sqrt.py')
-rw-r--r-- | config-model/src/test/cfg/application/ml_serving/models/sqrt.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/config-model/src/test/cfg/application/ml_serving/models/sqrt.py b/config-model/src/test/cfg/application/ml_serving/models/sqrt.py new file mode 100644 index 00000000000..b7b99b3850c --- /dev/null +++ b/config-model/src/test/cfg/application/ml_serving/models/sqrt.py @@ -0,0 +1,23 @@ +# Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. + +import onnx +from onnx import helper, TensorProto + +INPUT = helper.make_tensor_value_info('input', TensorProto.FLOAT, [1]) +OUTPUT = helper.make_tensor_value_info('out/layer/1:1', TensorProto.FLOAT, [1]) + +nodes = [ + helper.make_node( + 'Sqrt', + ['input'], + ['out/layer/1:1'], + ), +] +graph_def = helper.make_graph( + nodes, + 'sqrt', + [INPUT], + [OUTPUT], +) +model_def = helper.make_model(graph_def, producer_name='sqrt.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) +onnx.save(model_def, 'sqrt.onnx') |