Searched refs:LeafModelOperator (Results 1 – 7 of 7) sorted by relevance
29 class LeafModelOperator {34 explicit LeafModelOperator(const TensorForestParams& params) in LeafModelOperator() function36 virtual ~LeafModelOperator() {} in ~LeafModelOperator()59 class DenseClassificationLeafModelOperator : public LeafModelOperator {63 : LeafModelOperator(params) {} in DenseClassificationLeafModelOperator()78 class SparseClassificationLeafModelOperator : public LeafModelOperator {82 : LeafModelOperator(params) {} in SparseClassificationLeafModelOperator()95 class SparseOrDenseClassificationLeafModelOperator : public LeafModelOperator {99 : LeafModelOperator(params), in SparseOrDenseClassificationLeafModelOperator()120 class RegressionLeafModelOperator : public LeafModelOperator {[all …]
28 using tensorflow::tensorforest::LeafModelOperator;65 void TestClassificationNormalUse(const std::unique_ptr<LeafModelOperator>& op) { in TestClassificationNormalUse()89 std::unique_ptr<LeafModelOperator> op( in TEST()97 std::unique_ptr<LeafModelOperator> op( in TEST()105 std::unique_ptr<LeafModelOperator> op( in TEST()136 std::unique_ptr<LeafModelOperator> op( in TEST()166 std::unique_ptr<LeafModelOperator> op( in TEST()199 std::unique_ptr<LeafModelOperator> op( in TEST()
22 std::unique_ptr<LeafModelOperator>27 return std::unique_ptr<LeafModelOperator>( in CreateLeafModelOperator()31 return std::unique_ptr<LeafModelOperator>( in CreateLeafModelOperator()35 return std::unique_ptr<LeafModelOperator>( in CreateLeafModelOperator()39 return std::unique_ptr<LeafModelOperator>( in CreateLeafModelOperator()
86 std::shared_ptr<LeafModelOperator> model_op_;
77 std::shared_ptr<LeafModelOperator> model_op_;
259 std::unique_ptr<LeafModelOperator> model_op_;372 std::unique_ptr<LeafModelOperator> model_op_;
417 const std::unique_ptr<LeafModelOperator>& leaf_op, in FinalizeLeaf()494 std::unique_ptr<LeafModelOperator> model_op_;