Searched refs:GetOutputValue (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
H A Dleaf_model_operators_test.cc70 EXPECT_EQ(op->GetOutputValue(l, i), 0);
83 EXPECT_FLOAT_EQ(op->GetOutputValue(l, 1), 3.4);
128 EXPECT_FLOAT_EQ(op->GetOutputValue(*leaf, 0), 1.1);
129 EXPECT_FLOAT_EQ(op->GetOutputValue(*leaf, 1), 2.2);
130 EXPECT_FLOAT_EQ(op->GetOutputValue(*leaf, 2), 3.3);
158 EXPECT_FLOAT_EQ(op->GetOutputValue(*leaf, 0), 1.1);
159 EXPECT_FLOAT_EQ(op->GetOutputValue(*leaf, 1), 2.2);
160 EXPECT_FLOAT_EQ(op->GetOutputValue(*leaf, 2), 3.3);
187 EXPECT_FLOAT_EQ(op->GetOutputValue(*leaf, 0), 1.1);
188 EXPECT_FLOAT_EQ(op->GetOutputValue(*lea
[all...]
H A Dleaf_model_operators.h41 virtual float GetOutputValue(const decision_trees::Leaf& leaf,
64 float GetOutputValue(const decision_trees::Leaf& leaf,
83 float GetOutputValue(const decision_trees::Leaf& leaf,
102 float GetOutputValue(const decision_trees::Leaf& leaf,
124 float GetOutputValue(const decision_trees::Leaf& leaf,
H A Dleaf_model_operators.cc49 float DenseClassificationLeafModelOperator::GetOutputValue( function in class:tensorflow::tensorforest::DenseClassificationLeafModelOperator
79 float SparseClassificationLeafModelOperator::GetOutputValue( function in class:tensorflow::tensorforest::SparseClassificationLeafModelOperator
113 float SparseOrDenseClassificationLeafModelOperator::GetOutputValue( function in class:tensorflow::tensorforest::SparseOrDenseClassificationLeafModelOperator
116 return dense_->GetOutputValue(leaf, o);
118 return sparse_->GetOutputValue(leaf, o);
141 float RegressionLeafModelOperator::GetOutputValue( function in class:tensorflow::tensorforest::RegressionLeafModelOperator
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
H A Dmodel_ops.cc245 const float count = model_op_->GetOutputValue(leaf, j);

Completed in 180 milliseconds