ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onnx-ops Namespace Reference

Functions

 _dump_operators (onnx_model)
 
 _help_exit (cmd_name)
 
 main ()
 

Function Documentation

◆ _dump_operators()

onnx-ops._dump_operators (   onnx_model)
protected

Definition at line 22 of file onnx-ops.py.

22def _dump_operators(onnx_model):
23 for node in onnx_model.graph.node:
24 print(node.op_type)
25
26

Referenced by main().

◆ _help_exit()

onnx-ops._help_exit (   cmd_name)
protected

Definition at line 27 of file onnx-ops.py.

27def _help_exit(cmd_name):
28 print('Dump ONNX model file Operators')
29 print('Usage: {0} [onnx_path]'.format(cmd_name))
30 print('')
31 exit()
32
33

Referenced by main().

◆ main()

onnx-ops.main ( void  )

Definition at line 34 of file onnx-ops.py.

34def main():
35 if len(sys.argv) < 2:
36 _help_exit(os.path.basename(sys.argv[0]))
37
38 onnx.checker.check_model(sys.argv[1])
39 onnx_model = onnx.load(sys.argv[1])
40
41 _dump_operators(onnx_model)
42
43
int main(void)

References _dump_operators(), _help_exit(), and main().

Referenced by main().