examples/research_projects/onnx/summarization/bart_onnx/generation_onnx.py/EncoderForONNX/forward
class EncoderForONNX:    def forward(self, input_ids, attention_mask):
        return self.encoder(
            input_ids=input_ids,
            attention_mask=attention_mask,
            return_dict=False,
        )