adding alternate method to apply mask to allow XLA to detect MHA patt…#18

abhinavgoel95 wants to merge 3 commits intogoogle:maingoogle/praxis:mainfrom

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