Home
last modified time | relevance | path

Searched defs:SDPBackend (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/attention/
H A D__init__.py7 from torch._C import _SDPBackend as SDPBackend unknown
37 SDPBackend = SDPBackend variable
71 def sdpa_kernel(backends: Union[List[SDPBackend], SDPBackend]):
/aosp_15_r20/external/pytorch/test/
H A Dtest_transformers.py1392 def test_invalid_fused_inputs_dim_3(self, device, kernel: SDPBackend):
1410 def test_invalid_fused_inputs_broadcast(self, device, kernel: SDPBackend):
1425 def test_invalid_sequence_lengths(self, device, kernel: SDPBackend):
1439 def test_invalid_last_dim_stride(self, device, kernel: SDPBackend):
1454 def test_invalid_fused_inputs_head_dim(self, device, kernel: SDPBackend):
1472 def test_invalid_fused_inputs_invalid_dtype(self, device, kernel: SDPBackend):
1484 def test_invalid_fused_inputs_attn_mask_present(self, device, kernel: SDPBackend):
1546 def test_invalid_inputs_different_datatypes(self, device, kernel: SDPBackend):
1557 def test_invalid_inputs_different_devices(self, device, kernel: SDPBackend):
1566 def test_invalid_inputs_1_dimensional_inputs(self, device, kernel: SDPBackend):
[all …]
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_attention.py64 self, is_causal: bool, compiled: bool, backend: SDPBackend
/aosp_15_r20/external/pytorch/torch/backends/cuda/
H A D__init__.py264 from torch._C import _SDPAParams as SDPAParams, _SDPBackend as SDPBackend unknown
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/
H A Dsdp_utils_cpp.h26 enum class SDPBackend { enum