onnx-si Namespace Reference

Functions
	_help_exit (cmd_name)
	main ()

Function Documentation

_help_exit()

onnx-si._help_exit (   cmd_name )

protected

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

def _help_exit(cmd_name):
    print('Produce shape-infered ONNX file')
    print('Usage: {0} [onnx_in_path] [onnx_out_path]'.format(cmd_name))
    print('')
    exit()
 
 

Referenced by main().

main()

onnx-si.main

(

void

)

Definition at line 29 of file onnx-si.py.

def main():
    if len(sys.argv) < 3:
        _help_exit(os.path.basename(sys.argv[0]))
 
    onnx.checker.check_model(sys.argv[1])
    onnx.shape_inference.infer_shapes_path(sys.argv[1], sys.argv[2])
 
 

References _help_exit(), and main().

Referenced by main().