Home
last modified time | relevance | path

Searched defs:set_residual_tile (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/iterators/
H A Dpredicated_tile_access_iterator_residual_last.h285 void set_residual_tile(bool is_residual_tile) { in set_residual_tile() function
581 void set_residual_tile(bool enable) { in set_residual_tile() function
804 void set_residual_tile(bool enable) { in set_residual_tile() function
1093 void set_residual_tile(bool is_residual_tile) { in set_residual_tile() function
1356 void set_residual_tile(bool enable) { in set_residual_tile() function
1575 void set_residual_tile(bool enable) { in set_residual_tile() function
1803 void set_residual_tile(bool enable) { in set_residual_tile() function
2030 void set_residual_tile(bool enable) { in set_residual_tile() function
H A Dpredicated_tile_iterator_residual_last.h343 void set_residual_tile(bool enable) { in set_residual_tile() function
622 void set_residual_tile(bool enable) { in set_residual_tile() function
852 void set_residual_tile(bool enable) { in set_residual_tile() function
1095 void set_residual_tile(bool enable) { in set_residual_tile() function
1369 void set_residual_tile(bool enable) { in set_residual_tile() function
1596 void set_residual_tile(bool enable) { in set_residual_tile() function
1838 void set_residual_tile(bool enable) { in set_residual_tile() function
2067 void set_residual_tile(bool enable) { in set_residual_tile() function