Home
last modified time | relevance | path

Searched refs:sum_indices_vec (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_conditional_accumulator.h249 std::vector<int64>* sum_indices_vec = new std::vector<int64>(); in AddToAccumGradFunction() local
250 sum_indices_vec->reserve(sum_nnz); in AddToAccumGradFunction()
280 sum_indices_vec->push_back(accum_idx_vec_->at(idx_a)); in AddToAccumGradFunction()
288 sum_indices_vec->push_back(accum_idx_vec_->at(idx_a)); in AddToAccumGradFunction()
297 sum_indices_vec->push_back(grad_idx->vec<int64>()(idx_b)); in AddToAccumGradFunction()
309 accum_idx_vec_ = sum_indices_vec; in AddToAccumGradFunction()