Searched refs:dropout_config_ (Results 1 – 2 of 2) sorted by relevance
127 dropout_config_ = learner_config.learning_rate_tuner().dropout(); in GradientTreesPredictionOp()259 seed, dropout_config_, trees_not_to_drop, in DoCompute()330 LearningRateDropoutDrivenConfig dropout_config_; member in tensorflow::boosted_trees::GradientTreesPredictionOp
313 dropout_config_ = learner_config_.learning_rate_tuner().dropout(); in GrowTreeEnsembleOp()588 dropout_seed, dropout_config_, trees_not_to_drop, weights, in UpdateTreeWeightsIfDropout()891 LearningRateDropoutDrivenConfig dropout_config_; member in tensorflow::boosted_trees::GrowTreeEnsembleOp