Searched defs:head_dims (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/test/inductor/ |
| D | test_flex_decoding.py | 387 self, dtype: torch.dtype, score_mod: Callable, head_dims argument
417 def test_strided_inputs(self, dtype: torch.dtype, k_s, v_s, head_dims): argument
558 def test_non_equal_head_dims(self, dtype, score_mod, head_dims): argument
|