Searched defs:k_data (Results 1 – 2 of 2) sorted by relevance
261 const scalar_t* k_data = key.const_data_ptr<scalar_t>(); in cpu_flash_attention() local524 const scalar_t* k_data = key.const_data_ptr<scalar_t>(); in cpu_flash_attention_backward() local
398 const scalar_t* k_data = key.const_data_ptr<scalar_t>(); in cpu_flash_attention() local