Searched refs:check_for_seq_len_0_nested_tensor (Results 1 – 2 of 2) sorted by relevance
581 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()
185 inline bool check_for_seq_len_0_nested_tensor(sdp_params const& params, bool debug) { in check_for_seq_len_0_nested_tensor() function