def METHOD_NAME(self) -> Mapping[str, Mapping[int, str]]:
    common_inputs = {
        "input_ids": {0: "batch", 1: "encoder_sequence"},
        "attention_mask": {0: "batch", 1: "encoder_sequence"},
    }
    if self.use_past:
        common_inputs["attention_mask"][1] = "past_encoder_sequence + sequence"
        common_inputs["decoder_input_ids"] = {0: "batch"}
        common_inputs["decoder_attention_mask"] = {0: "batch", 1: "past_decoder_sequence + sequence"}
    else:
        common_inputs["decoder_input_ids"] = {0: "batch", 1: "decoder_sequence"}
        common_inputs["decoder_attention_mask"] = {0: "batch", 1: "decoder_sequence"}
    if self.use_past:
        self.fill_with_past_key_values_(common_inputs, direction="inputs")
    return common_inputs