aimet_tensorflow.cross_layer_equalization

Top-level API

aimet_tensorflow.keras.cross_layer_equalization.equalize_model(model)[source]

High-level API to perform Cross-Layer Equalization (CLE) on the given model

Parameters:

model (Model) – tf.keras.Model

Return type:

Model

Returns:

CLE applied tf.keras.Model