src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py/TFWav2Vec2GroupNorm/_check_axis
class TFWav2Vec2GroupNorm:    def _check_axis(self):

        if self.axis == 0:
            raise ValueError(
                "You are trying to normalize your batch axis. Do you want to "
                "use tf.layer.batch_normalization instead"
            )