src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py/FlaxConvWithWeightNorm/__call__
class FlaxConvWithWeightNorm:    def __call__(self, hidden_states):
        kernel = self._get_normed_weights()
        hidden_states = jnp.pad(hidden_states, ((0, 0), (self.prev_padding, self.prev_padding), (0, 0)))
        hidden_states = self.conv.apply({"params": {"kernel": kernel.T, "bias": self.bias}}, hidden_states)
        return hidden_states