Searched refs:_exp_reduce_sum_fusion_kernel (Results 1 – 2 of 2) sorted by relevance
90 _exp_reduce_sum_fusion_kernel(T1* a, const int& size, T2* out, T1& val) { in _exp_reduce_sum_fusion_kernel() function536 _exp_reduce_sum_fusion_kernel( in cpu_flash_attention()
55 inline void _exp_reduce_sum_fusion_kernel( in _exp_reduce_sum_fusion_kernel() function361 _exp_reduce_sum_fusion_kernel( in cpu_flash_attention()