Add alternate method to apply mask to allow XLA to detect MHA pattern#27

ashors1 wants to merge 1 commit intogoogle:maingoogle/praxis:mainfrom

nvjax-svc-0:patch/alternate-masking-methodnvjax-svc-0/praxis:patch/alternate-masking-methodCopy head branch name to clipboard