Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc673 auto* dst_vec = dest->mutable_vector(); in MergeLeafWeights() local
674 CHECK(src_vec.value_size() == dst_vec->value_size()); in MergeLeafWeights()
676 (*dst_vec->mutable_value()->Mutable(idx)) += src_vec.value(idx); in MergeLeafWeights()
691 auto* dst_vec = dest->mutable_sparse_vector(); in MergeLeafWeights() local
692 CHECK(dst_vec->value_size() == dst_vec->index_size()); in MergeLeafWeights()
693 dst_map.reserve(dst_vec->value_size()); in MergeLeafWeights()
694 for (size_t idx = 0; idx < dst_vec->value_size(); ++idx) { in MergeLeafWeights()
695 dst_map[dst_vec->index(idx)] = dst_vec->value(idx); in MergeLeafWeights()
702 dst_vec->clear_index(); in MergeLeafWeights()
703 dst_vec->clear_value(); in MergeLeafWeights()
[all …]