adding alternate method to apply mask to allow XLA to detect MHA pattern#19

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

abhinavgoel95:patch/mask-pattern-for-xlaabhinavgoel95/praxis:patch/mask-pattern-for-xlaCopy head branch name to clipboard