src/transformers/models/groupvit/modeling_tf_groupvit.py/TFGroupViTMixerMLP/call
class TFGroupViTMixerMLP:    def call(self, x, training: bool = False):
        x = super().call(hidden_states=tf.transpose(x, perm=(0, 2, 1)))
        return tf.transpose(x, perm=(0, 2, 1))