(multi_net_def, model_conf)
| 59 | |
| 60 | |
| 61 | def add_input_output_tensor(multi_net_def, model_conf): |
| 62 | if len(model_conf[ModelKeys.subgraphs]) == 1: |
| 63 | model_conf = model_conf[ModelKeys.subgraphs][ModelKeys.default_graph] |
| 64 | for input_tensor in model_conf[ModelKeys.input_tensors]: |
| 65 | multi_net_def.input_tensor.append(input_tensor) |
| 66 | for output_tensor in model_conf[ModelKeys.output_tensors]: |
| 67 | multi_net_def.output_tensor.append(output_tensor) |
| 68 | |
| 69 | |
| 70 | def convert(conf, output, enable_micro=False): |