src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py/FlaxWav2Vec2AdapterLayer/__call__
class FlaxWav2Vec2AdapterLayer:    def __call__(self, hidden_states):
        hidden_states = self.conv(hidden_states)
        hidden_states = nn.glu(hidden_states, axis=2)

        return hidden_states