TrainingExtensions/tensorflow/src/python/aimet_tensorflow/keras/utils/common.py
def _remove_network_nodes(model: tf.keras.Model, layers: typing.List[tf.keras.layers.Layer]):
    """
    Remove network nodes in the model corresponding to the given layers
    :param model: Model to remove network nodes from
    :param layers: List of layers corresponding to network nodes to remove
    """
    # pylint: disable=protected-access
    for layer in layers:
        node_key = layer.name + '_ib-0'
        model._network_nodes.remove(node_key)

    Remove network nodes in the model corresponding to the given layers
    :param model: Model to remove network nodes from
    :param layers: List of layers corresponding to network nodes to remove
    
Remove network nodes in the model corresponding to the given layers
Remove network nodes in the model corresponding to the given layers
def _remove_network_nodes(model: tf.keras.Model, layers: typing.List[tf.keras.layers.Layer]):
    for layer in layers:
        node_key = layer.name + '_ib-0'
        model._network_nodes.remove(node_key)
def _remove_network_nodes ( model : tf . keras . Model , layers : typing . List [ tf . keras . layers . Layer ] ) : for layer in layers : node_key = layer . name + '_ib-0' model . _network_nodes . remove ( node_key )
Remove network nodes in the model corresponding to the given layers
Remove network nodes in the model corresponding to the given layers
"""
    Remove network nodes in the model corresponding to the given layers
    :param model: Model to remove network nodes from
    :param layers: List of layers corresponding to network nodes to remove
    """ # pylint: disable=protected-access
model
tf.keras.Model
Model to remove network nodes from
Model to remove network nodes from
layers
typing.List[tf.keras.layers.Layer]
List of layers corresponding to network nodes to remove
List of layers corresponding to network nodes to remove