Home
last modified time | relevance | path

Searched refs:check_for_seq_len_0_nested_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/
H A Dsdp_utils.cpp581 check_for_seq_len_0_nested_tensor); in can_use_flash_attention()
643 check_for_seq_len_0_nested_tensor); in can_use_mem_efficient_attention()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/
H A Dsdp_utils_cpp.h185 inline bool check_for_seq_len_0_nested_tensor(sdp_params const& params, bool debug) { in check_for_seq_len_0_nested_tensor() function