diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-19 09:42:14 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-19 09:46:36 -0700 |
commit | 8d172e1a89feb06f906de43d75f0d5e65a2e1a04 (patch) | |
tree | bae824448fd7971200e7acc6af8dbb25ba8a2a67 /tensorflow/contrib/tensor_forest | |
parent | 98f522d3e982daafa9ccf136894cc83f496f5a11 (diff) |
Refactor code to improve TensorDataSet construction speed.
PiperOrigin-RevId: 189595482
Diffstat (limited to 'tensorflow/contrib/tensor_forest')
-rw-r--r-- | tensorflow/contrib/tensor_forest/kernels/v4/input_data.h | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tensorflow/contrib/tensor_forest/kernels/v4/input_data.h b/tensorflow/contrib/tensor_forest/kernels/v4/input_data.h index c544a8c75e..b991e6339f 100644 --- a/tensorflow/contrib/tensor_forest/kernels/v4/input_data.h +++ b/tensorflow/contrib/tensor_forest/kernels/v4/input_data.h @@ -44,12 +44,15 @@ class TensorDataSet { int column_count = 0; for (int i = 0; i < input_spec_.dense_size(); ++i) { for (int j = 0; j < input_spec_.dense(i).size(); ++j) { - decision_trees::FeatureId id; - id.mutable_id()->set_value(strings::StrCat(column_count)); - available_features_.push_back(id); ++column_count; } } + available_features_.reserve(column_count); + decision_trees::FeatureId id; + for (int i = 0; i < column_count; i++) { + id.mutable_id()->set_value(strings::StrCat(i)); + available_features_.emplace_back(id); + } // Set up the random number generator. if (split_sampling_random_seed_ == 0) { |