Home
last modified time | relevance | path

Searched refs:collection_op_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dfertile-stats-resource.cc26 if (collection_op_->IsInitialized(node_id)) { in AddExampleToStatsAndInitialize()
27 collection_op_->AddExample(input_data, target, examples, node_id); in AddExampleToStatsAndInitialize()
32 collection_op_->CreateAndInitializeCandidateWithExample( in AddExampleToStatsAndInitialize()
34 if (collection_op_->IsInitialized(node_id)) { in AddExampleToStatsAndInitialize()
40 *is_finished = collection_op_->IsFinished(node_id); in AddExampleToStatsAndInitialize()
44 collection_op_->InitializeSlot(node_id, depth); in AllocateNode()
56 collection_op_->ClearSlot(node); in Clear()
61 return collection_op_->BestSplit(node_id, best, depth); in BestSplit()
65 collection_op_->MaybeInitialize(); in MaybeInitialize()
69 collection_op_ = in ExtractFromProto()
[all …]
Dfertile-stats-resource.h55 collection_op_->ClearSlot(node_id); in ResetSplitStats()
56 collection_op_->InitializeSlot(node_id, depth); in ResetSplitStats()
87 std::unique_ptr<SplitCollectionOperator> collection_op_; variable