Searched refs:spmm_reduce_kernel (Results 1 – 1 of 1) sorted by relevance
430 void spmm_reduce_kernel( in spmm_reduce_kernel() function537 spmm_reduce_kernel(grad_other, ccol_indices, row, val, grad_out, ReductionType::SUM); in spmm_reduce_backward_other_kernel()558 REGISTER_DISPATCH(spmm_reduce_stub, &spmm_reduce_kernel);