Searched refs:original_weights (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
H A Ddropout_utils_test.cc56 std::vector<float> original_weights; local
67 &original_weights));
71 EXPECT_TRUE(original_weights.empty());
81 &original_weights));
85 EXPECT_EQ(kNumTrees, original_weights.size());
86 EXPECT_EQ(original_weights, weights_);
103 &original_weights));
122 std::vector<float> original_weights; local
134 &original_weights));
138 EXPECT_EQ(kNumTrees, original_weights
239 std::vector<float> original_weights; local
[all...]
H A Ddropout_utils.cc39 std::vector<float>* original_weights) {
44 if (original_weights == nullptr) {
62 original_weights->clear();
102 original_weights->push_back(weights[dropped_tree]);
35 DropOutTrees( const uint64 seed, const LearningRateDropoutDrivenConfig& config, const std::unordered_set<int32>& trees_not_to_drop, const std::vector<float>& weights, std::vector<int32>* dropped_trees, std::vector<float>* original_weights) argument
H A Ddropout_utils.h46 std::vector<float>* original_weights);
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
H A Dprediction_ops.cc229 std::vector<float> original_weights; local
252 weights, &dropped_trees, &original_weights));
308 output_dropout_info(1, i) = original_weights[i];

Completed in 176 milliseconds