Searched refs:model_op_ (Results 1 – 5 of 5) sorted by relevance
26 model_op_ = LeafModelOperatorFactory::CreateLeafModelOperator(params_); in DecisionTreeResource()64 model_op_->ExportModel(best->left_stats(), left_leaf); in SplitNode()71 model_op_->ExportModel(best->right_stats(), right_leaf); in SplitNode()86 model_op_->InitModel(tree->add_nodes()->mutable_leaf()); in MaybeInitialize()
40 model_op_ = LeafModelOperatorFactory::CreateLeafModelOperator(params_); in FertileStatsResource()86 std::shared_ptr<LeafModelOperator> model_op_; variable
77 std::shared_ptr<LeafModelOperator> model_op_; variable
172 model_op_ = LeafModelOperatorFactory::CreateLeafModelOperator(param_proto_); in TreePredictionsV4Op()245 const float count = model_op_->GetOutputValue(leaf, j); in set_output_value()259 std::unique_ptr<LeafModelOperator> model_op_; member in tensorflow::tensorforest::TreePredictionsV4Op331 model_op_ = LeafModelOperatorFactory::CreateLeafModelOperator(param_proto_); in UpdateModelV4Op()364 model_op_->UpdateModel( in UpdateModel()372 std::unique_ptr<LeafModelOperator> model_op_; member in tensorflow::tensorforest::UpdateModelV4Op
462 model_op_ = LeafModelOperatorFactory::CreateLeafModelOperator(param_proto_); in FinalizeTreeOp()487 param_proto_.drop_final_class(), model_op_, in Compute()494 std::unique_ptr<LeafModelOperator> model_op_; member in tensorflow::tensorforest::FinalizeTreeOp