Searched defs:sdpa_hop (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/inductor/ |
| D | test_flex_decoding.py | 375 def sdpa_hop(q, k, v, score_mod, block_mask): function
891 def sdpa_hop(q, k, v, score_mod): function
|
| D | test_flex_attention.py | 849 def sdpa_hop(q, k, v, score_mod): function
|