Searched defs:q_num_heads (Results 1 – 2 of 2) sorted by relevance
214 auto q_num_heads = params.query.size(1); in check_for_seq_len_0_nested_tensor() local347 const auto q_num_heads = params.query.sym_size(-3); in check_grouped_query_attention() local397 auto q_num_heads = params.query.sym_size(-3); in check_batch_size_and_num_heads_dense() local
222 const int64_t q_num_heads = query.size(1); in sdpa_nested_preprocessing_with_broadcast() local396 const int64_t q_num_heads = query.size(1); in sdpa_nested_preprocessing() local473 const int64_t q_num_heads = query.size(1); in sdpa_nested_preprocessing_backward() local