Searched refs:DecisionTreeEnsembleResource (Results 1 – 5 of 5) sorted by relevance
29 using boosted_trees::models::DecisionTreeEnsembleResource;47 auto* result = new boosted_trees::models::DecisionTreeEnsembleResource(); in Compute()72 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()91 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()115 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()149 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()195 REGISTER_RESOURCE_HANDLE_KERNEL(DecisionTreeEnsembleResource);199 IsResourceInitialized<boosted_trees::models::DecisionTreeEnsembleResource>);
50 using boosted_trees::models::DecisionTreeEnsembleResource;166 DecisionTreeEnsembleResource* ensemble_resource; in Compute()188 DecisionTreeEnsembleResource* ensemble_resource, in DoCompute()356 DecisionTreeEnsembleResource* ensemble_resource, in DoCompute()375 DecisionTreeEnsembleResource* ensemble_resource; in Compute()391 DecisionTreeEnsembleResource* ensemble_resource) { in DoCompute()
196 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()259 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource, in RetrieveBias()322 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()562 boosted_trees::models::DecisionTreeEnsembleResource* const in UpdateTreeWeightsIfDropout()613 boosted_trees::models::DecisionTreeEnsembleResource* const in UpdateAndRetrieveGrowableTree()723 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource) { in SplitTreeNode()771 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource) { in SplitTreeLayer()906 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute()
29 class DecisionTreeEnsembleResource : public StampedResource {32 explicit DecisionTreeEnsembleResource() in DecisionTreeEnsembleResource() function
24 REGISTER_RESOURCE_HANDLE_OP(DecisionTreeEnsembleResource);