src/transformers/models/data2vec/modeling_tf_data2vec_vision.py/TFData2VecVisionUperHead/psp_forward
class TFData2VecVisionUperHead:    def psp_forward(self, inputs):
        x = inputs[-1]
        psp_outs = [x]
        psp_outs.extend(self.psp_modules(x))
        psp_outs = tf.concat(psp_outs, axis=-1)
        output = self.bottleneck(psp_outs)

        return output