src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py/FlaxWav2Vec2FeatureEncoder/__call__
class FlaxWav2Vec2FeatureEncoder:    def __call__(self, input_values, freeze_feature_encoder=False):
        hidden_states = input_values[:, :, None]
        hidden_states = self.conv_layers(hidden_states)
        if freeze_feature_encoder:
            hidden_states = jax.lax.stop_gradient(hidden_states)
        return hidden_states