(output_config, flags, conf)
| 75 | |
| 76 | |
| 77 | def gen_sub_model_conf(output_config, flags, conf): |
| 78 | model_conf = copy.deepcopy(get_model_conf_by_name(flags, conf)) |
| 79 | if isinstance(model_conf[ModelKeys.subgraphs], list): |
| 80 | model_conf[ModelKeys.subgraphs][0][ |
| 81 | ModelKeys.output_tensors] = output_config[ModelKeys.output_tensors] |
| 82 | model_conf[ModelKeys.subgraphs][0][ |
| 83 | ModelKeys.output_shapes] = output_config[ModelKeys.output_shapes] |
| 84 | else: |
| 85 | model_conf[ModelKeys.subgraphs][ModelKeys.default_graph][ |
| 86 | ModelKeys.output_tensors] = output_config[ModelKeys.output_tensors] |
| 87 | model_conf[ModelKeys.subgraphs][ModelKeys.default_graph][ |
| 88 | ModelKeys.output_shapes] = output_config[ModelKeys.output_shapes] |
| 89 | return model_conf |
| 90 | |
| 91 | |
| 92 | def run_layers_validate(flags, args, original_conf): |
no test coverage detected