def _add_models_prefix(self): """Adds model name prefix to node names""" if not self._prefix_is_applied: self._prefix_is_applied = True for model_dict in self._models: model_name, model = model_dict['name'], model_dict['model'] for node in ge.get_all_operation_nodes(model, recursively=False): node.name = f'{model_name}_{node.name}'
def _add_models_prefix(self): """Adds model name prefix to node names""" if not self._prefix_is_applied: self._prefix_is_applied = True for model_dict in self._models: model_name, model = model_dict['name'], model_dict['model'] for node in ge.get_all_operation_nodes(model): node.name = '{}_{}'.format(model_name, node.name)
def _restore_models_prefix(self): """Restores removed model name prefix in node name""" if self._cache.node_names: self._prefix_is_applied = True for model_dict in self._models: model_name, model = model_dict['name'], model_dict['model'] for node in ge.get_all_operation_nodes(model, recursively=False): if node.name in self._cache.node_names[model_name]: rename_node(node, f'{model_name}_{node.name}') self._cache.pop('node_names')
def _restore_models_prefix(self): """Restores removed model name prefix in node name""" if self._cache.node_names: self._prefix_is_applied = True for model_dict in self._models: model_name, model = model_dict['name'], model_dict['model'] for node in ge.get_all_operation_nodes(model): if node.name in self._cache.node_names[model_name]: node.name = '{}_{}'.format(model_name, node.name) self._cache.pop('node_names')
def _remove_models_prefix(self): """Removes model name prefix from node names""" if self._prefix_is_applied: self._prefix_is_applied = False for model_dict in self._models: model_name, model = model_dict['name'], model_dict['model'] self._cache.node_names[model_name] = [] for node in ge.get_all_operation_nodes(model, recursively=False): if node.name.startswith(model_name): rename_node(node, node.name.replace(model_name + '_', '', 1)) self._cache.node_names[model_name].append(node.name)