Searched refs:check_for_seq_len_1_nested_tensor (Results 1 – 3 of 3) sorted by relevance
10 bool check_for_seq_len_1_nested_tensor(sdp_params const& params, bool debug);
733 bool check_for_seq_len_1_nested_tensor(sdp_params const& params, bool debug) { in check_for_seq_len_1_nested_tensor() function
567 …bool no_seq_len_1_nested = query.is_nested() ? check_for_seq_len_1_nested_tensor(kernel_params, fa… in native_multi_head_attention_cuda()