diff options
Diffstat (limited to 'config-model/src/test/cfg/application/onnx_name_collision/models/foobar.py')
-rwxr-xr-x | config-model/src/test/cfg/application/onnx_name_collision/models/foobar.py | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/config-model/src/test/cfg/application/onnx_name_collision/models/foobar.py b/config-model/src/test/cfg/application/onnx_name_collision/models/foobar.py new file mode 100755 index 00000000000..c3e0667aeaa --- /dev/null +++ b/config-model/src/test/cfg/application/onnx_name_collision/models/foobar.py @@ -0,0 +1,26 @@ +# 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('foo', TensorProto.FLOAT, [1]) +INPUT_2 = helper.make_tensor_value_info('baz', TensorProto.FLOAT, [1]) +OUTPUT = helper.make_tensor_value_info('bar', TensorProto.FLOAT, [1]) + +nodes = [ + helper.make_node( + 'Mul', + ['foo', 'baz'], + ['bat'], + ), +] +graph_def = helper.make_graph( + nodes, + 'mul', + [ + INPUT_1, + INPUT_2 + ], + [OUTPUT], +) +model_def = helper.make_model(graph_def, producer_name='foobar.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) +onnx.save(model_def, 'foobar.onnx') |