Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/
Dsdp_utils_cpp.cpp16 const auto value_size_last = params.value.sym_size(-1); in check_head_dim_size_cpp() local
/external/pytorch/aten/src/ATen/native/transformers/cuda/
Dsdp_utils.cpp87 const auto value_size_last = params.value.sym_size(-1); in check_head_dim_size_flash() local
111 const auto value_size_last = params.value.sym_size(-1); in check_head_dim_size_flash_nested() local
134 const auto value_size_last = params.value.sym_size(-1); in check_head_dim_size_mem_efficient() local