mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
fdce4fa6af
Description: Format all python files under onnxruntime with black and isort. After checking in, we can use .git-blame-ignore-revs to ignore the formatting PR in git blame. #11315, #11316
22 lines
950 B
Python
22 lines
950 B
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
|
|
import onnx
|
|
from onnx import TensorProto, helper
|
|
from onnx.helper import make_opsetid
|
|
|
|
input_info = helper.make_tensor_value_info("input", TensorProto.BFLOAT16, [1, 5])
|
|
output_info = helper.make_tensor_value_info("output", TensorProto.BFLOAT16, [1, 5])
|
|
|
|
# Create a node (NodeProto) - This is based on Pad-11
|
|
node_def = helper.make_node("Identity", ["input"], ["output"]) # node name # inputs # outputs
|
|
|
|
graph_def = helper.make_graph(nodes=[node_def], name="test_types_BLOAT16", inputs=[input_info], outputs=[output_info])
|
|
|
|
model_def = helper.make_model(graph_def, producer_name="AIInfra", opset_imports=[make_opsetid("", 13)])
|
|
|
|
onnx.checker.check_model(model_def)
|
|
onnx.helper.strip_doc_string(model_def)
|
|
final_model = onnx.shape_inference.infer_shapes(model_def)
|
|
onnx.checker.check_model(final_model)
|
|
onnx.save(final_model, "test_types_BFLOAT16.onnx")
|