Searched refs:GetFeatureFnType (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils.h | 36 using GetFeatureFnType = std::function<float(int32, int32)>; variable
157 GetFeatureFnType GetDenseFunctor(const Tensor& dense);
161 GetFeatureFnType GetSparseFunctor(const Tensor& sparse_indices,
170 bool DecideNode(const GetFeatureFnType& get_dense,
171 const GetFeatureFnType& get_sparse, int32 i, int32 feature,
|
| D | tree_utils.cc | 539 GetFeatureFnType GetDenseFunctor(const Tensor& dense) { in GetDenseFunctor()
555 GetFeatureFnType GetSparseFunctor(const Tensor& sparse_indices, in GetSparseFunctor()
573 bool DecideNode(const GetFeatureFnType& get_dense, in DecideNode()
574 const GetFeatureFnType& get_sparse, int32 i, int32 feature, in DecideNode()
|