Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/
H A Dsdp_utils_cpp.cpp42 check_nested_tensor, in use_flash_attention_cpp()
H A Dsdp_utils_cpp.h235 inline bool check_nested_tensor(sdp_params const& params, bool debug) { in check_nested_tensor() function