Searched defs:use_nt (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_native_mha.py | 18 self, device, dtype, use_nt, use_padding=False argument
113 …self, device, dtype, mode, use_nt, need_weights, average_attn_weights, use_padding=False, pad_all=… argument
277 def test_native_multihead_self_attention(self, device, dtype, use_nt, argument
|