mirror of
https://github.com/luxfi/onnx-source-snapshot.git
synced 2026-10-30 15:23:38 +00:00
756 B
756 B
onnx.tools
net_drawer
.. autofunction:: onnx.tools.net_drawer.GetPydotGraph
.. autofunction:: onnx.tools.net_drawer.GetOpNodeProducer
from onnx.tools.net_drawer import GetPydotGraph, GetOpNodeProducer
pydot_graph = GetPydotGraph(
model_onnx.graph, # model_onnx is a ModelProto instance
name=model_onnx.graph.name,
rankdir="TP",
node_producer=GetOpNodeProducer("docstring"))
pydot_graph.write_dot("graph.dot")
update_inputs_outputs_dims
.. autofunction:: onnx.tools.update_model_dims.update_inputs_outputs_dims
replace_initializer_by_constant_of_shape
.. autofunction:: onnx.tools.replace_constants.replace_initializer_by_constant_of_shape