Lines Matching refs:model_name
139 for model_name in tqdm(model_names):
140 print(datetime.datetime.now(), 'Processing', model_name)
141 html_data += '<h3>{}</h3>'.format(model_name)
142 model_data = ModelData(nnapi_model_name=model_name, manager=self)
151 def generate_hist_animation_html(self, model_name): argument
153 html_data = '<h3>{}</h3>'.format(model_name)
154 model_data = ModelData(nnapi_model_name=model_name, manager=self)
157 print(datetime.datetime.now(), "Done histogram for", model_name)
158 self.return_dict[model_name + "-hist"] = html_data
160 def generate_heatmap_animation_html(self, model_name): argument
162 model_data = ModelData(nnapi_model_name=model_name, manager=self)
165 print(datetime.datetime.now(), "Done heatmap for", model_name)
166 self.return_dict[model_name + "-heatmap"] = html_data
178 for model_name in model_names:
180 p = multiprocessing.Process(target=target_func, args=(model_name,))
188 for model_name in model_names:
189 f.write(self.return_dict[model_name + "-hist"])
190 f.write(self.return_dict[model_name + "-heatmap"])
425 def __init__(self, dump_dir_list, android_build_top, tflite_model_json_dir, model_name): argument
429 self.set_model_name(model_name)
431 def set_model_name(self, model_name): argument
433 self.model_name = model_name
444 model_data = ModelData(nnapi_model_name=self.model_name, manager=manager)
534 if args.no_parallel or args.model_name:
539 if args.model_name:
540 model_data = ModelData(nnapi_model_name=model_name, manager=manager)
542 generation_func(output_file_path=output_file_path, model_names=[args.model_name])