Searched refs:load_with_pointer_offset (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/iterators/ |
| H A D | predicated_tile_iterator_residual_last.h | 366 void load_with_pointer_offset(Fragment& frag, Index pointer_offset) { in load_with_pointer_offset() function
646 void load_with_pointer_offset(Fragment& frag, Index pointer_offset) { in load_with_pointer_offset() function
647 iterator_.load_with_pointer_offset(frag, pointer_offset); in load_with_pointer_offset()
659 load_with_pointer_offset(frag, 0); in load()
876 void load_with_pointer_offset(Fragment& frag, Index pointer_offset) { in load_with_pointer_offset() function
877 iterator_.load_with_pointer_offset(frag, pointer_offset); in load_with_pointer_offset()
889 load_with_pointer_offset(frag, 0); in load()
1118 void load_with_pointer_offset(Fragment& frag, Index pointer_offset) { in load_with_pointer_offset() function
1393 void load_with_pointer_offset(Fragment& frag, Index pointer_offset) { in load_with_pointer_offset() function
1394 iterator_.load_with_pointer_offset(frag, pointer_offset); in load_with_pointer_offset()
[all …]
|