Searched refs:DropOutTrees (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | dropout_utils_test.cc | 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()
132 TF_EXPECT_OK(DropoutUtils::DropOutTrees(kSeed, config, trees_not_to_drop, in TEST_F()
153 TF_EXPECT_OK(DropoutUtils::DropOutTrees(kSeed, config, trees_not_to_drop, in TEST_F()
171 TF_EXPECT_OK(DropoutUtils::DropOutTrees(kSeed, config, trees_not_to_drop, in TEST_F()
201 TF_EXPECT_OK(DropoutUtils::DropOutTrees( in TEST_F()
204 TF_EXPECT_OK(DropoutUtils::DropOutTrees( in TEST_F()
225 TF_EXPECT_OK(DropoutUtils::DropOutTrees(kSeed, config, trees_not_to_drop, in TEST_F()
228 TF_EXPECT_OK(DropoutUtils::DropOutTrees(kSeed, config, trees_not_to_drop, in TEST_F()
[all …]
|
| D | dropout_utils.h | 42 static tensorflow::Status DropOutTrees(
|
| D | dropout_utils.cc | 35 Status DropoutUtils::DropOutTrees( in DropOutTrees() function in tensorflow::boosted_trees::utils::DropoutUtils
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | prediction_ops.cc | 258 OP_REQUIRES_OK(context, DropoutUtils::DropOutTrees( in DoCompute()
|
| D | training_ops.cc | 587 const auto dropout_status = DropoutUtils::DropOutTrees( in UpdateTreeWeightsIfDropout()
|