diff options
author | Lester Solbakken <lesters@oath.com> | 2021-05-19 11:35:40 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2021-05-19 11:35:40 +0200 |
commit | a186020aa62214a714f24091b7928a159a55b166 (patch) | |
tree | 418641c48b1fde584c19b8914608fee00bd37628 /model-integration/src/test/models/onnx/add_double.py | |
parent | 00a724c605b3d1332a119454f1382830df2226d2 (diff) |
Add ONNX-RT evaluator to model-integration module
Diffstat (limited to 'model-integration/src/test/models/onnx/add_double.py')
-rwxr-xr-x | model-integration/src/test/models/onnx/add_double.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/model-integration/src/test/models/onnx/add_double.py b/model-integration/src/test/models/onnx/add_double.py new file mode 100755 index 00000000000..fa9aa48f4b2 --- /dev/null +++ b/model-integration/src/test/models/onnx/add_double.py @@ -0,0 +1,27 @@ +# 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_1 = helper.make_tensor_value_info('input1', TensorProto.DOUBLE, [1]) +INPUT_2 = helper.make_tensor_value_info('input2', TensorProto.DOUBLE, [1]) +OUTPUT = helper.make_tensor_value_info('output', TensorProto.DOUBLE, [1]) + +nodes = [ + helper.make_node( + 'Add', + ['input1', 'input2'], + ['output'], + ), +] +graph_def = helper.make_graph( + nodes, + 'add', + [ + INPUT_1, + INPUT_2 + ], + [OUTPUT], +) +model_def = helper.make_model(graph_def, producer_name='add_double.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) +onnx.save(model_def, 'add_double.onnx') |