src/transformers/models/deit/modeling_tf_deit.py/TFDeitDecoder/call
class TFDeitDecoder:    def call(self, inputs: tf.Tensor, training: bool = False) -> tf.Tensor:
        hidden_states = inputs
        hidden_states = self.conv2d(hidden_states)
        hidden_states = self.pixel_shuffle(hidden_states)
        return hidden_states