Get rid of mask related warnings when using MHA layer with mask #3999

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Check the code format	1m 49s
Run tests (3.9, tensorflow)	15m 18s
Run tests (3.9, jax)	20m 32s
Run tests (3.9, torch)	14m 25s
Run tests (3.9, numpy)	6m 47s
	58m 51s