(model_infos, input_dir)
| 31 | |
| 32 | |
| 33 | def parse_model_info(model_infos, input_dir): |
| 34 | input_list = [] |
| 35 | model_data_info = [] |
| 36 | model_data_info_local = [] |
| 37 | model_shape_info = [] |
| 38 | for model_info in model_infos[0]: |
| 39 | model_input_info = [] |
| 40 | model_input_info_local = [] |
| 41 | model_input_shape_info = [] |
| 42 | for shape, dtype, input_name in model_info: |
| 43 | input_file_name = gen_input_name(shape, dtype) |
| 44 | input_path = os.path.join(input_dir, input_file_name) |
| 45 | input_list.append({ |
| 46 | "input_file_name": input_file_name, |
| 47 | "input_path": input_path, |
| 48 | "shape": shape, |
| 49 | "dtype": dtype, |
| 50 | "input_name": input_name, |
| 51 | }) |
| 52 | model_input_info.append(input_name + "=" + input_file_name) |
| 53 | model_input_info_local.append(input_name + "=" + input_path) |
| 54 | model_input_shape_info.append(input_name + "=(" + |
| 55 | ",".join([str(i) |
| 56 | for i in shape]) + ")") |
| 57 | model_data_info.append(";".join(model_input_info)) |
| 58 | model_shape_info.append(";".join(model_input_shape_info)) |
| 59 | model_data_info_local.append(";".join(model_input_info_local)) |
| 60 | model_data_info = ":".join(model_data_info) |
| 61 | model_shape_info = ":".join(model_shape_info) |
| 62 | model_data_info_local = ":".join(model_data_info_local) |
| 63 | |
| 64 | output_name_2_dtype = {} |
| 65 | for output_info in model_infos[1]: |
| 66 | for _, dtype, name in output_info: |
| 67 | output_name_2_dtype[name] = dtype |
| 68 | return input_list, model_data_info, model_shape_info, model_data_info_local, output_name_2_dtype |
| 69 | |
| 70 | |
| 71 | def local_call(cmd): |
no test coverage detected