11_onnx_build Namespace Reference

Variables
	X = helper.make_tensor_value_info('X', TensorProto.FLOAT, [1, 1, 64, 64])
	W_info = helper.make_tensor_value_info('W', TensorProto.FLOAT, [2, 1, 3, 3])
	W = np.array([[[[-1,-1,-1],[-1,8,-1],[-1,-1,-1]]],[[[0,-1,0],[-1,5,-1],[0,-1,0]]]]).astype(np.float32)
	Y = helper.make_tensor_value_info('Y', TensorProto.FLOAT, [1, 2, 64, 64])
	conv1
	graph_def
	model_def = helper.make_model(graph_def, producer_name='benchmarks')

Variable Documentation

conv1

11_onnx_build.conv1

Initial value:

=  helper.make_node(
        'Conv',         # name
        ['X', 'W'],     # inputs
        ['Y'],          # outputs
        
        kernel_shape=[3, 3],
        strides=[1, 1],
        pads=[1, 1, 1, 1],
        )

Definition at line 19 of file 11_onnx_build.py.

graph_def

11_onnx_build.graph_def

Initial value:

=  helper.make_graph(
        nodes=[conv1],     # graph nodes
        name= 'conv2d_b_model',   # graph name
        inputs =      [X, W_info],  # graph inputs
        outputs =     [Y],          # graph outputs
        initializer = [W],
        )

Definition at line 30 of file 11_onnx_build.py.

model_def

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

Definition at line 38 of file 11_onnx_build.py.

W

11_onnx_build.W = np.array([[[[-1,-1,-1],[-1,8,-1],[-1,-1,-1]]],[[[0,-1,0],[-1,5,-1],[0,-1,0]]]]).astype(np.float32)

Definition at line 12 of file 11_onnx_build.py.

W_info

11_onnx_build.W_info = helper.make_tensor_value_info('W', TensorProto.FLOAT, [2, 1, 3, 3])

Definition at line 11 of file 11_onnx_build.py.

X

11_onnx_build.X = helper.make_tensor_value_info('X', TensorProto.FLOAT, [1, 1, 64, 64])

Definition at line 9 of file 11_onnx_build.py.

Y

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

Definition at line 17 of file 11_onnx_build.py.