Searched refs:SplitCollectionOperator (Results 1 – 5 of 5) sorted by relevance
28 REGISTER_SPLIT_COLLECTION(COLLECTION_BASIC, SplitCollectionOperator);30 std::unique_ptr<SplitCollectionOperator>43 std::unique_ptr<GrowStats> SplitCollectionOperator::CreateGrowStats( in CreateGrowStats()68 void SplitCollectionOperator::ExtractFromProto( in ExtractFromProto()77 void SplitCollectionOperator::PackToProto(FertileStats* stats_proto) const { in PackToProto()88 void SplitCollectionOperator::InitializeSlot(int32 node_id, int32 depth) { in InitializeSlot()93 void SplitCollectionOperator::AddExample( in AddExample()102 bool SplitCollectionOperator::IsInitialized(int32 node_id) const { in IsInitialized()111 void SplitCollectionOperator::CreateAndInitializeCandidateWithExample( in CreateAndInitializeCandidateWithExample()139 bool SplitCollectionOperator::BestSplit(int32 node_id, SplitCandidate* best, in BestSplit()
34 class SplitCollectionOperator {36 explicit SplitCollectionOperator(const TensorForestParams& params) in SplitCollectionOperator() function38 virtual ~SplitCollectionOperator() {} in ~SplitCollectionOperator()96 virtual std::unique_ptr<SplitCollectionOperator> Create(103 static std::unique_ptr<SplitCollectionOperator> CreateSplitCollectionOperator(115 virtual std::unique_ptr<SplitCollectionOperator> Create( in Create()117 return std::unique_ptr<SplitCollectionOperator>(new T(params)); in Create()
34 class GraphRunnerSplitCollectionOperator : public SplitCollectionOperator {37 : SplitCollectionOperator(params) { in GraphRunnerSplitCollectionOperator()
87 std::unique_ptr<SplitCollectionOperator> collection_op_;
135 SplitCollectionOperator::ClearSlot(node_id); in ClearSlot()