12_onnx_build Namespace Reference

Variables
	X = helper.make_tensor_value_info('X', TensorProto.FLOAT, [1, 1, 8, 8])
	Y = helper.make_tensor_value_info('Y', TensorProto.FLOAT, [1, 1, 2, 2])
	maxp1
	graph_def
	model_def = helper.make_model(graph_def, producer_name='benchmarks')

Variable Documentation

graph_def

12_onnx_build.graph_def

Initial value:

=  helper.make_graph(
        nodes=[maxp1],     # graph nodes
        name= 'maxp_a_model',   # graph name
        inputs =      [X],  # graph inputs
        outputs =     [Y],          # graph outputs
        )

Definition at line 25 of file 12_onnx_build.py.

maxp1

12_onnx_build.maxp1

Initial value:

=  helper.make_node(
        'MaxPool',         # name
        ['X'],     # inputs
        ['Y'],          # outputs
        
        auto_pad = 'VALID',
        kernel_shape=[3, 3],
        strides=[3, 3],
        )

Definition at line 14 of file 12_onnx_build.py.

model_def

12_onnx_build.model_def = helper.make_model(graph_def, producer_name='benchmarks')

Definition at line 32 of file 12_onnx_build.py.

X

12_onnx_build.X = helper.make_tensor_value_info('X', TensorProto.FLOAT, [1, 1, 8, 8])

Definition at line 9 of file 12_onnx_build.py.

Y

12_onnx_build.Y = helper.make_tensor_value_info('Y', TensorProto.FLOAT, [1, 1, 2, 2])

Definition at line 12 of file 12_onnx_build.py.