Searched refs:current_weights (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | dropout_utils_test.cc | 329 std::vector<float> current_weights = weights; in TEST_F() local
331 std::vector<int32>(current_weights.size(), 1); in TEST_F()
333 dropped_1, GetWeightsByIndex(current_weights, dropped_1), in TEST_F()
334 current_weights.size(), 1, ¤t_weights, &num_updates); in TEST_F()
336 dropped_2, GetWeightsByIndex(current_weights, dropped_2), in TEST_F()
337 current_weights.size(), 1, ¤t_weights, &num_updates); in TEST_F()
338 res_1 = current_weights; in TEST_F()
342 std::vector<float> current_weights = weights; in TEST_F() local
344 std::vector<int32>(current_weights.size(), 1); in TEST_F()
347 dropped_2, GetWeightsByIndex(current_weights, dropped_2), in TEST_F()
[all …]
|
| D | dropout_utils.cc | 112 std::vector<float>* current_weights, std::vector<int32>* num_updates) { in GetTreesWeightsForAddingTrees() argument
113 CHECK(num_updates->size() == current_weights->size()); in GetTreesWeightsForAddingTrees()
127 if (new_tree_index < current_weights->size()) { in GetTreesWeightsForAddingTrees()
129 (*current_weights)[new_tree_index] = in GetTreesWeightsForAddingTrees()
134 current_weights->push_back(total_new_trees_weight / num_trees_to_add); in GetTreesWeightsForAddingTrees()
143 (*current_weights)[dropped] = new_weight; in GetTreesWeightsForAddingTrees()
|
| D | dropout_utils.h | 68 std::vector<float>* current_weights,
|