xla

Fix op test for `nn.functional.scaled_dot_product_attention"nn.functi…

#8025

Merged