exclude test/
This commit is contained in:
-19
@@ -1,19 +0,0 @@
|
||||
import onnx
|
||||
from onnx import TensorProto, helper
|
||||
|
||||
graph = helper.make_graph(
|
||||
[ # nodes
|
||||
helper.make_node("Add", ["A", "B"], ["C"], "Add"),
|
||||
],
|
||||
"SingleAdd", # name
|
||||
[ # inputs
|
||||
helper.make_tensor_value_info("A", TensorProto.FLOAT, [1]),
|
||||
helper.make_tensor_value_info("B", TensorProto.FLOAT, [1]),
|
||||
],
|
||||
[ # outputs
|
||||
helper.make_tensor_value_info("C", TensorProto.FLOAT, [1]),
|
||||
],
|
||||
)
|
||||
|
||||
model = helper.make_model(graph, opset_imports=[helper.make_opsetid("", 12)])
|
||||
onnx.save(model, r"single_add.onnx")
|
||||
Reference in New Issue
Block a user