[llama-mm] Add torch.cond to replace if condition in MHA#6869

larryliu0820 merged 2 commits intomainpytorch/executorch:mainfrom

torch_cond_attentionpytorch/executorch:torch_cond_attentionCopy head branch name to clipboard