Respect the "inputs_bidirection_attention" setting when creating the cache prefilling mask for DecoderOnly prompted models.
Respect the "inputs_bidirection_attention" setting when creating the cache prefilling mask for DecoderOnly prompted models.

PiperOrigin-RevId: 436259615

google-research/prompt-tuning,google-research/prompt-tuning