Go to the source code of this file.
|
| 05_onnx_build.X = helper.make_tensor_value_info('X', TensorProto.FLOAT, [1]) |
|
| 05_onnx_build.W_info = helper.make_tensor_value_info('W', TensorProto.FLOAT, [1,64]) |
|
| 05_onnx_build.W = np.ones((1,64)).astype(np.float32) |
|
| 05_onnx_build.B_info = helper.make_tensor_value_info('B', TensorProto.FLOAT, [64]) |
|
| 05_onnx_build.B = np.ones((64)).astype(np.float32) |
|
| 05_onnx_build.Z = helper.make_tensor_value_info('Z', TensorProto.FLOAT, [64]) |
|
| 05_onnx_build.matmul1 |
|
| 05_onnx_build.bias1 |
|
| 05_onnx_build.graph_def |
|
| 05_onnx_build.model_def = helper.make_model(graph_def, producer_name='benchmarks') |
|