read.rst 1.55 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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
.. option::  --optimize, -O

Optimize when reading

.. 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.

85
86
87
88
.. option::  --py

Print out program using python API.

89
90
91
92
.. option::  --output, -o [std::string]

Output to file.