07_onnx_build Namespace Reference

Variables
	X = helper.make_tensor_value_info('X', TensorProto.FLOAT, [1,64])
	Z = helper.make_tensor_value_info('Z', TensorProto.FLOAT, [1,64])
	smax1
	graph_def
	model_def = helper.make_model(graph_def, producer_name='benchmarks')

Variable Documentation

graph_def

07_onnx_build.graph_def

Initial value:

=  helper.make_graph(
        nodes=[smax1],     # graph nodes
        name= 'softmax_b_model',   # graph name
        inputs =      [X],  # graph inputs
        outputs =     [Z],          # graph outputs
        )

Definition at line 21 of file 07_onnx_build.py.

model_def

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

Definition at line 28 of file 07_onnx_build.py.

smax1

07_onnx_build.smax1

Initial value:

=  helper.make_node(
        'Softmax',         # name
        ['X'],     # inputs
        ['Z'],          # outputs
        )

Definition at line 15 of file 07_onnx_build.py.

X

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

Definition at line 9 of file 07_onnx_build.py.

Z

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

Definition at line 13 of file 07_onnx_build.py.