src/transformers/models/gpt_neo/configuration_gpt_neo.py/GPTNeoOnnxConfig/inputs
class GPTNeoOnnxConfig:    def inputs(self) -> Mapping[str, Mapping[int, str]]:
        common_inputs = OrderedDict({"input_ids": {0: "batch", 1: "sequence"}})
        if self.use_past:
            for i in range(self._config.num_layers):
                common_inputs[f"past_key_values.{i}.key"] = {0: "batch", 2: "past_sequence"}
                common_inputs[f"past_key_values.{i}.value"] = {0: "batch", 2: "past_sequence"}

            common_inputs["attention_mask"] = {0: "batch", 1: "past_sequence + sequence"}
        else:
            common_inputs["attention_mask"] = {0: "batch", 1: "sequence"}

        return common_inputs