Searched refs:accum_idx_vec_ (Results 1 – 1 of 1) sorted by relevance
57 accum_idx_vec_ = nullptr; in SparseConditionalAccumulator()64 if (accum_idx_vec_ != nullptr) delete accum_idx_vec_; in ~SparseConditionalAccumulator() local71 std::vector<int64>* accum_idx_vec_ = nullptr;167 if (accum_idx_vec_ != nullptr) delete accum_idx_vec_; in AllocateAndAssignToAccumGradFunction() local168 accum_idx_vec_ = new std::vector<int64>(); in AllocateAndAssignToAccumGradFunction()169 accum_idx_vec_->reserve(nnz); in AllocateAndAssignToAccumGradFunction()171 accum_idx_vec_->push_back(grad_idx->vec<int64>()(i)); in AllocateAndAssignToAccumGradFunction()200 const int64 accum_nnz = accum_idx_vec_->size(); in AddToAccumGradFunction()219 switch (cmp(accum_idx_vec_, grad_idx, i, j)) { in AddToAccumGradFunction()280 sum_indices_vec->push_back(accum_idx_vec_->at(idx_a)); in AddToAccumGradFunction()[all …]