Home
last modified time | relevance | path

Searched refs:check_tensor_shapes (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/
H A Dsdp_utils_cpp.cpp44 check_tensor_shapes, in use_flash_attention_cpp()
H A Dsdp_utils_cpp.h312 inline bool check_tensor_shapes(sdp_params const& params, bool debug) { in check_tensor_shapes() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/
H A Dsdp_utils.cpp564 check_tensor_shapes, in can_use_flash_attention()
622 check_tensor_shapes, in can_use_mem_efficient_attention()