src/transformers/models/hubert/modeling_tf_hubert.py/TFHubertGroupNorm/_get_reshaped_weights
class TFHubertGroupNorm:    def _get_reshaped_weights(self, input_shape):
        broadcast_shape = self._create_broadcast_shape(input_shape)
        gamma = None
        beta = None
        if self.scale:
            gamma = tf.reshape(self.gamma, broadcast_shape)

        if self.center:
            beta = tf.reshape(self.beta, broadcast_shape)
        return gamma, beta