File: tools.md

package info (click to toggle)
onnx 1.17.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, trixie
  • size: 52,856 kB
  • sloc: python: 73,992; cpp: 53,539; makefile: 50; sh: 48; javascript: 1
file content (34 lines) | stat: -rw-r--r-- 756 bytes parent folder | download | duplicates (2)
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
```