diff options
author | Lester Solbakken <lesters@oath.com> | 2020-04-03 11:29:43 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2020-04-03 11:29:43 +0200 |
commit | 3789127189224d6cbd6f109b9a95f848869ea6cc (patch) | |
tree | 79cef74e6c61da059ed0eae79632fa001433ddc2 /model-integration/src/test/models/onnx/simple/concat.py | |
parent | 706cb2d3b2d623318ba9c0a8db0e4355448af65a (diff) |
for testing onlylesters/bert-testing
Diffstat (limited to 'model-integration/src/test/models/onnx/simple/concat.py')
-rwxr-xr-x | model-integration/src/test/models/onnx/simple/concat.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/model-integration/src/test/models/onnx/simple/concat.py b/model-integration/src/test/models/onnx/simple/concat.py new file mode 100755 index 00000000000..b77cf5decc1 --- /dev/null +++ b/model-integration/src/test/models/onnx/simple/concat.py @@ -0,0 +1,25 @@ +# Copyright 2020 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +import onnx +import numpy as np +from onnx import helper, TensorProto + +i_type = helper.make_tensor_value_info('i', TensorProto.FLOAT, [1]) +j_type = helper.make_tensor_value_info('j', TensorProto.FLOAT, [1]) +k_type = helper.make_tensor_value_info('k', TensorProto.FLOAT, [1]) + +output_type = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3]) + +node = onnx.helper.make_node( + 'Concat', + inputs=['i', 'j', 'k'], + outputs=['y'], + axis=0, +) +graph_def = onnx.helper.make_graph( + nodes = [node], + name = 'concat_test', + inputs = [i_type, j_type, k_type], + outputs = [output_type] +) +model_def = helper.make_model(graph_def, producer_name='concat.py') +onnx.save(model_def, 'concat.onnx') |