(filename)
| 80 | |
| 81 | |
| 82 | def read_pb_file(filename): |
| 83 | with open(filename, 'rb') as pfile: |
| 84 | data_str = pfile.read() |
| 85 | tensor = onnx.TensorProto() |
| 86 | tensor.ParseFromString(data_str) |
| 87 | np_array = numpy_helper.to_array(tensor) |
| 88 | |
| 89 | return tensor.name, np_array |
| 90 | |
| 91 | |
| 92 | def wrapup_inputs(io_folder, param_names): |
no test coverage detected