Searched defs:test_sdpa (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_flop_counter.py | 307 def test_sdpa(self): member in TestFlopCounter |
| H A D | test_decomp.py | 1179 def test_sdpa(self, device, dtype, op): member in DecompOneOffTests |
| H A D | test_nestedtensor.py | 6194 def test_sdpa(self, device, dtype): member in TestNestedTensorSubclass |
| /aosp_15_r20/external/pytorch/test/functorch/ | ||
| H A D | test_vmap.py | 3873 def test_sdpa(self, device, backend): member in TestVmapBatchedGradient |
| /aosp_15_r20/external/pytorch/test/inductor/ | ||
| H A D | test_aot_inductor.py | 954 def test_sdpa(self): member in AOTInductorTestsTemplate |
| H A D | test_torchinductor.py | 9019 def test_sdpa(self): member in CommonTemplate |