src/transformers/models/big_bird/modeling_flax_big_bird.py/FlaxBigBirdForQuestionAnsweringHead/__call__
class FlaxBigBirdForQuestionAnsweringHead:    def __call__(self, encoder_output, deterministic=True):
        hidden_states = self.dropout(encoder_output, deterministic=deterministic)
        hidden_states = self.intermediate(hidden_states)
        hidden_states = self.output(hidden_states, encoder_output)
        hidden_states = self.qa_outputs(hidden_states)
        return hidden_states