Home
last modified time | relevance | path

Searched refs:convert_flash_attn_S_to_softmax (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_transformers.py2108 def convert_flash_attn_S_to_softmax( member in TestSDPACudaOnly
2928 softmax_mask = self.convert_flash_attn_S_to_softmax(
3016 softmax_mask = self.convert_flash_attn_S_to_softmax(
3373 softmax_mask = self.convert_flash_attn_S_to_softmax(