src/transformers/models/segformer/modeling_tf_segformer.py/TFSegformerDWConv/__init__
class TFSegformerDWConv:    def __init__(self, dim: int = 768, **kwargs):
        super().__init__(**kwargs)
        self.depthwise_convolution = tf.keras.layers.Conv2D(
            filters=dim, kernel_size=3, strides=1, padding="same", groups=dim, name="dwconv"
        )