read.rst 1.5 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")

49
50
51
52
53
54
55
56
.. options:: --dyn-input-dim [std::vector<std::string>]

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

.. options:: --default-dyn-dim

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
85
86
87
88
.. 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.

.. option::  --output, -o [std::string]

Output to file.