Searched defs:max_seqlen_batch_k (Results 1 – 3 of 3) sorted by relevance
71 int64_t max_seqlen_batch_k, in _flash_attention_backward()199 const int64_t max_seqlen_batch_k = key.size(2); in _scaled_dot_product_cudnn_attention_backward_cuda() local742 const int64_t max_seqlen_batch_k, in _scaled_dot_product_flash_attention_backward_cuda()
699 const int64_t max_seqlen_batch_k = key.size(2); in _scaled_dot_product_flash_attention_cuda() local756 const int64_t max_seqlen_batch_k = key.size(2); in _scaled_dot_product_cudnn_attention_cuda() local850 int64_t max_seqlen_batch_k, in _flash_attention_forward()
336 const int64_t max_seqlen_batch_k, in _scaled_dot_product_flash_attention_backward_nested()