Searched defs:buf_reduced_data (Results 1 – 2 of 2) sorted by relevance
269 scalar_t* buf_reduced_data = is_reduced_type ? buf_reduced.data_ptr<scalar_t>() : nullptr; in cpu_flash_attention() local532 scalar_t* buf_reduced_data = is_reduced_type ? buf_reduced.data_ptr<scalar_t>() : nullptr; in cpu_flash_attention_backward() local
404 scalar_t* buf_reduced_data = in cpu_flash_attention() local