Searched defs:max_seqlen_batch_q (Results 1 – 4 of 4) sorted by relevance
70 int64_t max_seqlen_batch_q, in _flash_attention_backward()198 const int64_t max_seqlen_batch_q = query.size(2); in _scaled_dot_product_cudnn_attention_backward_cuda() local741 const int64_t max_seqlen_batch_q, in _scaled_dot_product_flash_attention_backward_cuda()
698 const int64_t max_seqlen_batch_q = query.size(2); in _scaled_dot_product_flash_attention_cuda() local753 const int64_t max_seqlen_batch_q = query.size(2); in _scaled_dot_product_cudnn_attention_cuda() local849 int64_t max_seqlen_batch_q, in _flash_attention_forward()
250 int64_t max_seqlen_batch_q = 0, Nnz_q = 0; in sdpa_nested_preprocessing_with_broadcast() local466 const int64_t max_seqlen_batch_q, in sdpa_nested_preprocessing_backward()
335 const int64_t max_seqlen_batch_q, in _scaled_dot_product_flash_attention_backward_nested()