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