def METHOD_NAME(self):
    input_ids = ids_tensor([self.batch_size, self.seq_length], self.text_vocab_size)
    input_mask = None
    if self.use_input_mask:
        input_mask = random_attention_mask([self.batch_size, self.seq_length])
    config = self.get_config()
    return config, input_ids, input_mask