src/transformers/models/groupvit/modeling_tf_groupvit.py/TFGroupViTCrossAttentionLayer/call
class TFGroupViTCrossAttentionLayer:    def call(self, query: tf.Tensor, key: tf.Tensor, training: bool = False) -> tf.Tensor:
        x = query
        x = x + self.attn(query, encoder_hidden_states=key)[0]
        x = x + self.mlp(self.norm2(x))
        x = self.norm_post(x)
        return x