src/transformers/models/xlnet/modeling_tf_xlnet.py/TFXLNetFeedForward/call
class TFXLNetFeedForward:    def call(self, inp, training=False):
        output = inp
        output = self.layer_1(output)
        output = self.activation_function(output)
        output = self.dropout(output, training=training)
        output = self.layer_2(output)
        output = self.dropout(output, training=training)
        output = self.layer_norm(output + inp)
        return output