(model_name, net_confs, net_def,
params, model_output)
| 47 | |
| 48 | |
| 49 | def convert_micro(model_name, net_confs, net_def, |
| 50 | params, model_output): |
| 51 | mace_check(len(net_confs.items()) == 1, |
| 52 | "Multi graph not supported in Micro") |
| 53 | for graph_name, graph_conf in net_confs.items(): |
| 54 | micro_converter = MicroConverter(graph_conf, copy.deepcopy(net_def), |
| 55 | copy.deepcopy(params), model_name) |
| 56 | micro_converter.gen_code() |
| 57 | micro_converter.package(model_output + "/" + |
| 58 | model_name + "_micro.tar.gz") |
| 59 | |
| 60 | |
| 61 | def add_input_output_tensor(multi_net_def, model_conf): |
no test coverage detected