Searched refs:center_bias_ (Results 1 – 2 of 2) sorted by relevance
98 OP_REQUIRES_OK(context, context->GetAttr("center_bias", ¢er_bias_)); in GradientTreesPredictionOp()249 if (center_bias_) { in DoCompute()268 only_finalized_trees_, center_bias_, &trees_to_include); in DoCompute()337 bool center_bias_; member in tensorflow::boosted_trees::GradientTreesPredictionOp
299 OP_REQUIRES_OK(context, context->GetAttr("center_bias", ¢er_bias_)); in GrowTreeEnsembleOp()575 if (center_bias_) { in UpdateTreeWeightsIfDropout()893 bool center_bias_; member in tensorflow::boosted_trees::GrowTreeEnsembleOp