Searched refs:model_dir (Results 1 – 1 of 1) sorted by relevance
122 def __init__(self, model_dir): argument125 dir_path = model_dir + useNNAPIDir + "/"216 self.model_dir = self.get_target_model_dir(manager.DUMP_DIR, nnapi_model_name)217 self.tensor_dict = TensorDict(self.model_dir)