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)