Searched refs:store_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 | 409 void store_with_pointer_offset(Fragment const& frag, Index pointer_offset) { in store_with_pointer_offset() function
664 void store_with_pointer_offset(Fragment const& frag, Index pointer_offset) { in store_with_pointer_offset() function
665 iterator_.store_with_pointer_offset(frag, pointer_offset); in store_with_pointer_offset()
677 store_with_pointer_offset(frag, 0); in store()
894 void store_with_pointer_offset(Fragment const& frag, Index pointer_offset) { in store_with_pointer_offset() function
895 iterator_.store_with_pointer_offset(frag, pointer_offset); in store_with_pointer_offset()
907 store_with_pointer_offset(frag, 0); in store()
1161 void store_with_pointer_offset(Fragment const& frag, Index pointer_offset) { in store_with_pointer_offset() function
1411 void store_with_pointer_offset(Fragment const& frag, Index pointer_offset) { in store_with_pointer_offset() function
1412 iterator_.store_with_pointer_offset(frag, pointer_offset); in store_with_pointer_offset()
[all …]
|