def METHOD_NAME(batch_size):
    return (
        (
            InputSpec((batch_size,), dtype="string"),
            InputSpec((batch_size,), dtype="string"),
        ),
        InputSpec(
            (batch_size, SEQUENCE_LENGTH), dtype="int64", range=[0, VOCAB_SIZE]
        ),
    )