def METHOD_NAME(self, module: tf.keras.layers.Layer):
    if isinstance(module, tf.keras.layers.Conv2D):
        params = aimet_common.layer_database.Conv2dTypeSpecificParams(module.strides, module.padding, module.groups)
        self.type_specific_params = params