Searched refs:FertileStatsResource (Results 1 – 5 of 5) sorted by relevance
22 void FertileStatsResource::AddExampleToStatsAndInitialize( in AddExampleToStatsAndInitialize()43 void FertileStatsResource::AllocateNode(int32 node_id, int32 depth) { in AllocateNode()47 void FertileStatsResource::Allocate(int32 parent_depth, in Allocate()55 void FertileStatsResource::Clear(int32 node) { in Clear()59 bool FertileStatsResource::BestSplit(int32 node_id, SplitCandidate* best, in BestSplit()64 void FertileStatsResource::MaybeInitialize() { in MaybeInitialize()68 void FertileStatsResource::ExtractFromProto(const FertileStats& stats) { in ExtractFromProto()74 void FertileStatsResource::PackToProto(FertileStats* stats) const { in PackToProto()
35 class FertileStatsResource : public ResourceBase {38 explicit FertileStatsResource(const TensorForestParams& params) in FertileStatsResource() function
56 auto* result = new FertileStatsResource(param_proto_); in Compute()90 FertileStatsResource* fertile_stats_resource; in Compute()119 FertileStatsResource* fertile_stats_resource; in Compute()148 void UpdateStats(FertileStatsResource* fertile_stats_resource, in UpdateStats()199 FertileStatsResource* fertile_stats_resource, in UpdateStatsCollated()254 FertileStatsResource* fertile_stats_resource; in Compute()365 FertileStatsResource* fertile_stats_resource; in Compute()469 FertileStatsResource* fertile_stats_resource; in Compute()498 REGISTER_RESOURCE_HANDLE_KERNEL(FertileStatsResource);501 IsResourceInitialized<FertileStatsResource>);
25 REGISTER_RESOURCE_HANDLE_OP(FertileStatsResource);
141 are stored in the `FertileStatsResource` defined in