src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py/TFWav2Vec2GroupNorm/_set_number_of_groups_for_instance_norm
class TFWav2Vec2GroupNorm:    def _set_number_of_groups_for_instance_norm(self, input_shape):
        dim = input_shape[self.axis]

        if self.groups == -1:
            self.groups = dim