from tensorflow.python.keras.utils.generic_utils import register_keras_serializable

    return outputs


@register_keras_serializable(package='Vitis', name='AveragePooling2D')
class VitisAveragePooling2D(tf.keras.layers.AveragePooling2D):
  """Vitis version of AveragePooling2D layer.
