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