diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-04-01 08:43:19 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-04-01 10:09:27 -0700 |
commit | 00d72581b3f71fe4cdffd861da84ad2c4bd31ba5 (patch) | |
tree | 201c0aee82bcb233317b4f90e8fb544556a5dad5 | |
parent | 643573d41f770961255707a1aa9413d54f979ba3 (diff) |
Fix reference to proto2 in boosted_trees.
Change: 151917679
-rw-r--r-- | tensorflow/contrib/boosted_trees/lib/testutil/random_tree_gen.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/boosted_trees/lib/testutil/random_tree_gen.cc b/tensorflow/contrib/boosted_trees/lib/testutil/random_tree_gen.cc index 24259ff603..cbe26ba918 100644 --- a/tensorflow/contrib/boosted_trees/lib/testutil/random_tree_gen.cc +++ b/tensorflow/contrib/boosted_trees/lib/testutil/random_tree_gen.cc @@ -30,8 +30,8 @@ namespace { // Append the given nodes to tree with transfer of pointer ownership. // nodes will not be usable upon return. -void AppendNodes(DecisionTreeConfig* tree, - proto2::RepeatedPtrField<TreeNode>* nodes) { +template <typename T> +void AppendNodes(DecisionTreeConfig* tree, T* nodes) { std::reverse(nodes->pointer_begin(), nodes->pointer_end()); while (!nodes->empty()) { tree->mutable_nodes()->AddAllocated(nodes->ReleaseLast()); |