e1_onnx_build Namespace Reference

Variables
	X = helper.make_tensor_value_info('X', TensorProto.FLOAT, [1,784])
	W_info = helper.make_tensor_value_info('W', TensorProto.FLOAT, [784,10])
	W = np.random.randn(784,10).astype(np.float32)
	B_info = helper.make_tensor_value_info('B', TensorProto.FLOAT, [1,10])
	B = np.ones([1,10]).astype(np.float32)
	Z = helper.make_tensor_value_info('Z', TensorProto.FLOAT, [1,10])
	matmul
	bias
	softmax
	graph_def
	model_def = helper.make_model(graph_def, producer_name='benchmarks')

Variable Documentation

B

e1_onnx_build.B = np.ones([1,10]).astype(np.float32)

Definition at line 18 of file e1_onnx_build.py.

B_info

e1_onnx_build.B_info = helper.make_tensor_value_info('B', TensorProto.FLOAT, [1,10])

Definition at line 17 of file e1_onnx_build.py.

bias

e1_onnx_build.bias

Initial value:

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

Definition at line 30 of file e1_onnx_build.py.

graph_def

e1_onnx_build.graph_def

Initial value:

=  helper.make_graph(
        nodes=[matmul, bias, softmax],     # graph nodes
        name= 'mlp_model',   # graph name
        inputs =      [X, W_info, B_info],  # graph inputs
        outputs =     [Z],          # graph outputs
        initializer = [W, B],
        )

Definition at line 44 of file e1_onnx_build.py.

matmul

e1_onnx_build.matmul

Initial value:

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

Definition at line 24 of file e1_onnx_build.py.

model_def

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

Definition at line 52 of file e1_onnx_build.py.

softmax

e1_onnx_build.softmax

Initial value:

=  helper.make_node(
        'Softmax',
        ['V'],
        ['Z']
        )

Definition at line 36 of file e1_onnx_build.py.

W

e1_onnx_build.W = np.random.randn(784,10).astype(np.float32)

Definition at line 14 of file e1_onnx_build.py.

W_info

e1_onnx_build.W_info = helper.make_tensor_value_info('W', TensorProto.FLOAT, [784,10])

Definition at line 13 of file e1_onnx_build.py.

X

e1_onnx_build.X = helper.make_tensor_value_info('X', TensorProto.FLOAT, [1,784])

Definition at line 11 of file e1_onnx_build.py.

Z

e1_onnx_build.Z = helper.make_tensor_value_info('Z', TensorProto.FLOAT, [1,10])

Definition at line 22 of file e1_onnx_build.py.