onnx_display_model.py File Reference

Namespaces
namespace	onnx_display_model

Functions
	onnx_display_model.get_shape (value_info)
	onnx_display_model.get_tensor_shape_by_name (model, name)

Variables
str	onnx_display_model.domain = opset.domain + ":" if opset.domain else ""
	onnx_display_model.end
	onnx_display_model.model = onnx.shape_inference.infer_shapes(model=onnx.load(sys.argv[1]), check_type=True, strict_mode=True)
	onnx_display_model.shape = get_tensor_shape_by_name(model, input_name)