e2_onnx_build.py File Reference

Go to the source code of this file.

Namespaces
	e2_onnx_build

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