diff options
Diffstat (limited to 'eval/src/tests/tensor/onnx_wrapper/simple.py')
-rwxr-xr-x | eval/src/tests/tensor/onnx_wrapper/simple.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/eval/src/tests/tensor/onnx_wrapper/simple.py b/eval/src/tests/tensor/onnx_wrapper/simple.py index a3cd2425d58..c8db58b5ebb 100755 --- a/eval/src/tests/tensor/onnx_wrapper/simple.py +++ b/eval/src/tests/tensor/onnx_wrapper/simple.py @@ -29,5 +29,6 @@ graph_def = helper.make_graph( ], [OUTPUT], ) -model_def = helper.make_model(graph_def, producer_name='simple.py') + +model_def = helper.make_model(graph_def, producer_name='simple.py', opset_imports=[onnx.OperatorSetIdProto(version=12)]) onnx.save(model_def, 'simple.onnx') |