1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
|
# onnx.tools
## net_drawer
```{eval-rst}
.. autofunction:: onnx.tools.net_drawer.GetPydotGraph
```
```{eval-rst}
.. 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
```{eval-rst}
.. autofunction:: onnx.tools.update_model_dims.update_inputs_outputs_dims
```
## replace_initializer_by_constant_of_shape
```{eval-rst}
.. autofunction:: onnx.tools.replace_constants.replace_initializer_by_constant_of_shape
```
|