(save_data_dir, runtime)
| 490 | return count |
| 491 | |
| 492 | def SaveInputsOneByOne(save_data_dir, runtime): |
| 493 | for i in range(runtime.GetInputCount()): |
| 494 | tensor = runtime.GetInputTensor(i) |
| 495 | shape = tensor.GetShape() |
| 496 | dims = shape.GetDims() |
| 497 | out_file_name = save_data_dir + "/pplnn_input_" + str(i) + "_" + tensor.GetName() + "-" + GenDimsStr(dims) + "-" + g_data_type_str[shape.GetDataType()] + ".dat" |
| 498 | element_count = CalcElementCount(dims) |
| 499 | if element_count > 0: |
| 500 | tensor_data = tensor.ConvertToHost() |
| 501 | if not tensor_data: |
| 502 | logging.error("copy data from tensor[" + tensor.GetName() + "] failed.") |
| 503 | sys.exit(-1) |
| 504 | |
| 505 | in_data = np.array(tensor_data, copy=False) |
| 506 | in_data.tofile(out_file_name) |
| 507 | else: |
| 508 | open(out_file_name, 'a').close() |
| 509 | |
| 510 | # ---------------------------------------------------------------------------- # |
| 511 |
no test coverage detected