Searched refs:original_weights (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
H A D | dropout_utils_test.cc | 56 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 D | dropout_utils.cc | 39 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 D | dropout_utils.h | 46 std::vector<float>* original_weights);
|
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
H A D | prediction_ops.cc | 229 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