05_onnx_build Namespace Reference

Variables
	X = helper.make_tensor_value_info('X', TensorProto.FLOAT, [1])
	W_info = helper.make_tensor_value_info('W', TensorProto.FLOAT, [1,64])
	W = np.ones((1,64)).astype(np.float32)
	B_info = helper.make_tensor_value_info('B', TensorProto.FLOAT, [64])
	B = np.ones((64)).astype(np.float32)
	Z = helper.make_tensor_value_info('Z', TensorProto.FLOAT, [64])
	matmul1
	bias1
	graph_def
	model_def = helper.make_model(graph_def, producer_name='benchmarks')

Variable Documentation

B

05_onnx_build.B = np.ones((64)).astype(np.float32)

Definition at line 16 of file 05_onnx_build.py.

B_info

05_onnx_build.B_info = helper.make_tensor_value_info('B', TensorProto.FLOAT, [64])

Definition at line 15 of file 05_onnx_build.py.

bias1

05_onnx_build.bias1

Initial value:

=  helper.make_node(
        'Add',
        ['Y', 'B'],
        ['Z'],
        )

Definition at line 28 of file 05_onnx_build.py.

graph_def

05_onnx_build.graph_def

Initial value:

=  helper.make_graph(
        nodes=[matmul1, bias1],     # graph nodes
        name= 'dense_b_model',   # graph name
        inputs =      [X, W_info, B_info],  # graph inputs
        outputs =     [Z],          # graph outputs
        initializer = [W, B],
        )

Definition at line 35 of file 05_onnx_build.py.

matmul1

05_onnx_build.matmul1

Initial value:

=  helper.make_node(
        'MatMul',         # name
        ['X', 'W'],     # inputs
        ['Y'],          # outputs
        )

Definition at line 22 of file 05_onnx_build.py.

model_def

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

Definition at line 43 of file 05_onnx_build.py.

W

05_onnx_build.W = np.ones((1,64)).astype(np.float32)

Definition at line 12 of file 05_onnx_build.py.

W_info

05_onnx_build.W_info = helper.make_tensor_value_info('W', TensorProto.FLOAT, [1,64])

Definition at line 11 of file 05_onnx_build.py.

X

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

Definition at line 9 of file 05_onnx_build.py.

Z

05_onnx_build.Z = helper.make_tensor_value_info('Z', TensorProto.FLOAT, [64])

Definition at line 20 of file 05_onnx_build.py.