aboutsummaryrefslogtreecommitdiffstats
path: root/model-integration/src/test/models/onnx/simple/const.py
diff options
context:
space:
mode:
Diffstat (limited to 'model-integration/src/test/models/onnx/simple/const.py')
-rwxr-xr-xmodel-integration/src/test/models/onnx/simple/const.py26
1 files changed, 26 insertions, 0 deletions
diff --git a/model-integration/src/test/models/onnx/simple/const.py b/model-integration/src/test/models/onnx/simple/const.py
new file mode 100755
index 00000000000..35d6dee1346
--- /dev/null
+++ b/model-integration/src/test/models/onnx/simple/const.py
@@ -0,0 +1,26 @@
+# 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
+
+output_type = helper.make_tensor_value_info('y', TensorProto.FLOAT, [])
+
+node = onnx.helper.make_node(
+ 'Constant',
+ inputs=[],
+ outputs=['y'],
+ value=onnx.helper.make_tensor(
+ name='const_tensor',
+ data_type=onnx.TensorProto.FLOAT,
+ dims=(),
+ vals=[0.42]
+ ),
+)
+graph_def = onnx.helper.make_graph(
+ nodes = [node],
+ name = 'constant_test',
+ inputs = [],
+ outputs = [output_type]
+)
+model_def = helper.make_model(graph_def, producer_name='const.py')
+onnx.save(model_def, 'const.onnx')