Searched refs:grad_nnz (Results 1 – 1 of 1) sorted by relevance
201 const int64 grad_nnz = grad_idx->dim_size(0); in AddToAccumGradFunction() local209 entries_to_copy.reserve(accum_nnz + grad_nnz); in AddToAccumGradFunction()217 while (i < accum_nnz && j < grad_nnz) { in AddToAccumGradFunction()242 while (j < grad_nnz) { in AddToAccumGradFunction()