Searched refs:learner_config_ (Results 1 – 1 of 1) sorted by relevance
186 learner_config_.ParseFromString(serialized_learner_config), in CenterTreeEnsembleBiasOp()284 boosted_trees::learner::LearnerConfig learner_config_; member in tensorflow::boosted_trees::CenterTreeEnsembleBiasOp306 learner_config_.ParseFromString(serialized_learner_config), in GrowTreeEnsembleOp()310 if (learner_config_.has_learning_rate_tuner() && in GrowTreeEnsembleOp()311 learner_config_.learning_rate_tuner().tuner_case() == in GrowTreeEnsembleOp()313 dropout_config_ = learner_config_.learning_rate_tuner().dropout(); in GrowTreeEnsembleOp()421 if (learner_config_.pruning_mode() == in Compute()472 UpdateBestSplit(learner_config_, partition_id, &split, best_splits); in FindBestSplitsPerPartitionNormal()523 if (learner_config_.pruning_mode() == learner::LearnerConfig::PRE_PRUNE && in FindBestSplitOblivious()650 if (learner_config_.growing_mode() == in MergeLeafWeights()[all …]