read.rst 1.61 KB
Newer Older
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
.. option::  <input file>

File to load

.. option::  --model [resnet50|inceptionv3|alexnet]

Load model

.. option::  --onnx

Load as onnx

.. option::  --tf

Load as tensorflow

.. option::  --migraphx

Load as MIGraphX

.. option::  --migraphx-json

Load as MIGraphX JSON

.. option::  --batch [unsigned int] (Default: 1)

27
For a static model, set batch size. For a dynamic batch model, sets the batch size at runtime.
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

.. option::  --nhwc

Treat tensorflow format as nhwc

.. option::  --skip-unknown-operators

Skip unknown operators when parsing and continue to parse.

.. option::  --nchw

Treat tensorflow format as nchw

.. option::  --trim, -t [unsigned int]

Trim instructions from the end (Default: 0)

.. option::  --input-dim [std::vector<std::string>]

Dim of a parameter (format: "@name d1 d2 dn")

Charlie Lin's avatar
Charlie Lin committed
49
.. option:: --dyn-input-dim [std::vector<std::string>]
50
51
52

Set dynamic dimensions of a parameter using JSON formatting (format "@name" "dynamic_dimension_json")

Charlie Lin's avatar
Charlie Lin committed
53
.. option:: --default-dyn-dim
54
55
56

Set the default dynamic dimension (format {min:x, max:y, optimals:[o1,o2,...]})

57
58
59
60
.. option::  --optimize, -O

Optimize when reading

61
62
63
64
.. option::  --apply-pass, -p

Passes to apply to model

65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
.. option::  --graphviz, -g

Print out a graphviz representation.

.. option::  --brief

Make the output brief.

.. option::  --cpp

Print out the program as cpp program.

.. option::  --json

Print out program as json.

.. option::  --text

Print out program in text format.

.. option::  --binary

Print out program in binary format.

89
90
91
92
.. option::  --py

Print out program using python API.

93
94
95
96
.. option::  --output, -o [std::string]

Output to file.