src/transformers/models/t5/modeling_tf_t5.py/TFT5LayerNorm/call
class TFT5LayerNorm:    def call(self, hidden_states):
        variance = tf.math.reduce_mean(tf.math.square(hidden_states), axis=-1, keepdims=True)
        hidden_states = hidden_states * tf.math.rsqrt(variance + self.variance_epsilon)
        return self.weight * hidden_states