Searched refs:num_queries_in_block (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/ |
| H A D | kernel_backward.h | 1478 int32_t num_queries_in_block = skipBoundsChecks in processBlockIJ() local
1491 {num_queries_in_block, p.head_dim_value - col}, in processBlockIJ()
1498 num_queries_in_block); in processBlockIJ()
1514 {num_queries_in_block, p.head_dim - col}, in processBlockIJ()
1521 num_queries_in_block); in processBlockIJ()
1527 {num_queries_in_block, p.head_dim_value}, in processBlockIJ()
1552 num_queries_in_block, in processBlockIJ()
1607 {num_queries_in_block, num_keys_in_block}, in processBlockIJ()
1788 num_keys_in_block, p.head_dim_value - col, num_queries_in_block); in processBlockIJ()
1799 {num_queries_in_block, p.head_dim_value - col}, in processBlockIJ()
[all …]
|