TrainingExtensions/tensorflow/src/python/aimet_tensorflow/keras/batch_norm_fold.py
def fold_all_batch_norms(model: (tf.keras.Model, tf.keras.layers.Layer)):
    """
    Fold all batch_norm layers in a model into corresponding conv layers

    :param model: model to find all batch norms for
    """

    bn_conv_linear_pairs = _find_all_batch_norms_to_fold(model)

    _fold_selected_batch_norms(model, bn_conv_linear_pairs)

    Fold all batch_norm layers in a model into corresponding conv layers

    :param model: model to find all batch norms for
    
Fold all batch_norm layers in a model into corresponding conv layers
Fold all batch_norm layers in a model into corresponding conv layers
def fold_all_batch_norms(model: (tf.keras.Model, tf.keras.layers.Layer)):
    bn_conv_linear_pairs = _find_all_batch_norms_to_fold(model)
    _fold_selected_batch_norms(model, bn_conv_linear_pairs)
def fold_all_batch_norms ( model : ( tf . keras . Model , tf . keras . layers . Layer ) ) : bn_conv_linear_pairs = _find_all_batch_norms_to_fold ( model ) _fold_selected_batch_norms ( model , bn_conv_linear_pairs )
Fold all batch_norm layers in a model into corresponding conv layers
Fold all batch_norm layers in a model into corresponding conv layers
"""
    Fold all batch_norm layers in a model into corresponding conv layers

    :param model: model to find all batch norms for
    """
model
(tf.keras.Model, tf.keras.layers.Layer)
model to find all batch norms for
model to find all batch norms for