Home
last modified time | relevance | path

Searched refs:trees_not_to_drop (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Ddropout_utils_test.cc57 std::unordered_set<int32> trees_not_to_drop; in TEST_F() local
65 TF_EXPECT_OK(DropoutUtils::DropOutTrees(kSeed, config, trees_not_to_drop, in TEST_F()
79 TF_EXPECT_OK(DropoutUtils::DropOutTrees(kSeed, config, trees_not_to_drop, in TEST_F()
101 TF_EXPECT_OK(DropoutUtils::DropOutTrees(seed, config, trees_not_to_drop, in TEST_F()
126 std::unordered_set<int32> trees_not_to_drop; in TEST_F() local
132 TF_EXPECT_OK(DropoutUtils::DropOutTrees(kSeed, config, trees_not_to_drop, in TEST_F()
144 std::unordered_set<int32> trees_not_to_drop; in TEST_F() local
146 trees_not_to_drop.insert(i); in TEST_F()
153 TF_EXPECT_OK(DropoutUtils::DropOutTrees(kSeed, config, trees_not_to_drop, in TEST_F()
163 std::unordered_set<int32> trees_not_to_drop; in TEST_F() local
[all …]
Ddropout_utils.cc37 const std::unordered_set<int32>& trees_not_to_drop, in DropOutTrees() argument
88 if (trees_not_to_drop.find(i) != trees_not_to_drop.end()) { in DropOutTrees()
Ddropout_utils.h44 const std::unordered_set<int32>& trees_not_to_drop,
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc248 std::unordered_set<int32> trees_not_to_drop; in DoCompute() local
250 trees_not_to_drop.insert(0); in DoCompute()
255 trees_not_to_drop.insert(ensemble_resource->num_trees() - 1); in DoCompute()
259 seed, dropout_config_, trees_not_to_drop, in DoCompute()
Dtraining_ops.cc574 std::unordered_set<int32> trees_not_to_drop; in UpdateTreeWeightsIfDropout() local
576 trees_not_to_drop.insert(0); in UpdateTreeWeightsIfDropout()
580 trees_not_to_drop.insert(current_tree); in UpdateTreeWeightsIfDropout()
588 dropout_seed, dropout_config_, trees_not_to_drop, weights, in UpdateTreeWeightsIfDropout()