Searched refs:check_tensor_shapes (Results 1 – 3 of 3) sorted by relevance
44 check_tensor_shapes, in use_flash_attention_cpp()
312 inline bool check_tensor_shapes(sdp_params const& params, bool debug) { in check_tensor_shapes() function
564 check_tensor_shapes, in can_use_flash_attention()622 check_tensor_shapes, in can_use_mem_efficient_attention()