Searched defs:buf_reduced (Results 1 – 2 of 2) sorted by relevance
257 …at::Tensor buf_reduced = at::empty({num_thread, qSplitSize, is_reduced_type ? kvSplitSize : 0}, qu… in cpu_flash_attention() local517 …at::Tensor buf_reduced = at::empty({num_thread, is_reduced_type ? size_per_thread_reduced : 0}, qu… in cpu_flash_attention_backward() local
391 void* buf_reduced = reinterpret_cast<void*>(buf_reduced_vec.data()); in cpu_flash_attention() local