src/transformers/models/data2vec/modeling_tf_data2vec_vision.py/TFAdaptiveAvgPool2D/__init__
class TFAdaptiveAvgPool2D:    def __init__(self, output_shape, mode="dense", **kwargs):
        super().__init__(**kwargs)
        self.mode = mode
        self.h_pool = TFAdaptiveAvgPool1D(output_shape[0], mode=mode, name="h_pool")
        self.w_pool = TFAdaptiveAvgPool1D(output_shape[1], mode=mode, name="w_pool")