onnx_display_model Namespace Reference

Functions
	get_shape (value_info)
	get_tensor_shape_by_name (model, name)

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

Function Documentation

get_shape()

onnx_display_model.get_shape

(

value_info

)

Here is the caller graph for this function:

get_tensor_shape_by_name()

onnx_display_model.get_tensor_shape_by_name	(		model,
			name
	)

Here is the call graph for this function:

Variable Documentation

domain

str onnx_display_model.domain = opset.domain + ":" if opset.domain else ""

end

onnx_display_model.end

model

onnx_display_model.model = onnx.shape_inference.infer_shapes(model=onnx.load(sys.argv[1]), check_type=True, strict_mode=True)

shape

onnx_display_model.shape = get_tensor_shape_by_name(model, input_name)