Searched defs:q_seq_len (Results 1 – 4 of 4) sorted by relevance
132 const int64_t q_seq_len = graph.size_at<int64_t>(1, q_projected); in get_cache_slice_sizes() local
782 auto q_seq_len = query.size(2); in flash_attention_kernel_out() local862 auto q_seq_len = q.size(1); in custom_sdpa_out() local
739 auto q_seq_len = query.size(2); in flash_attention_kernel_impl() local794 auto q_seq_len = query.size(1); in flash_attention_backward_kernel_impl() local
1891 q_seq_len, argument