Home
last modified time | relevance | path

Searched refs:tensorforest (Results 1 – 25 of 51) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Ddecision_node_evaluator_test.cc25 using tensorflow::tensorforest::InequalityDecisionNodeEvaluator;
26 using tensorflow::tensorforest::MatchingValuesDecisionNodeEvaluator;
27 using tensorflow::tensorforest::ObliqueInequalityDecisionNodeEvaluator;
37 std::unique_ptr<tensorflow::tensorforest::TensorDataSet> dataset( in TEST()
38 new tensorflow::tensorforest::TestableDataSet( in TEST()
54 std::unique_ptr<tensorflow::tensorforest::TensorDataSet> dataset( in TEST()
55 new tensorflow::tensorforest::TestableDataSet( in TEST()
71 std::unique_ptr<tensorflow::tensorforest::TensorDataSet> dataset( in TEST()
72 new tensorflow::tensorforest::TestableDataSet( in TEST()
88 std::unique_ptr<tensorflow::tensorforest::TensorDataSet> dataset( in TEST()
[all …]
Dgrow_stats_test.cc30 using tensorflow::tensorforest::DenseClassificationGrowStats;
31 using tensorflow::tensorforest::FertileSlot;
32 using tensorflow::tensorforest::FixedSizeClassStats;
33 using tensorflow::tensorforest::FixedSizeSparseClassificationGrowStats;
34 using tensorflow::tensorforest::GrowStats;
35 using tensorflow::tensorforest::LeastSquaresRegressionGrowStats;
36 using tensorflow::tensorforest::SparseClassificationGrowStats;
37 using tensorflow::tensorforest::SPLIT_FINISH_BASIC;
38 using tensorflow::tensorforest::SPLIT_FINISH_DOMINATE_HOEFFDING;
39 using tensorflow::tensorforest::SPLIT_PRUNE_HOEFFDING;
[all …]
Dleaf_model_operators_test.cc27 using tensorflow::tensorforest::DenseClassificationLeafModelOperator;
28 using tensorflow::tensorforest::LeafModelOperator;
29 using tensorflow::tensorforest::LeafStat;
30 using tensorflow::tensorforest::RegressionLeafModelOperator;
31 using tensorflow::tensorforest::SparseClassificationLeafModelOperator;
32 using tensorflow::tensorforest::SparseOrDenseClassificationLeafModelOperator;
33 using tensorflow::tensorforest::TensorForestParams;
34 using tensorflow::tensorforest::TestableInputTarget;
Dinput_data.h31 namespace tensorforest {
39 TensorDataSet(const tensorforest::TensorForestDataSpec& input_spec, in TensorDataSet()
118 const tensorforest::TensorForestDataSpec input_spec_;
Dparams_test.cc21 using tensorflow::tensorforest::DepthDependentParam;
22 using tensorflow::tensorforest::ResolveParam;
Dinput_data.cc21 namespace tensorforest { namespace
127 const int32 num_sparse = tensorforest::GetNumSparseFeatures( in RandomSample()
Dparams.h22 namespace tensorforest {
Dstat_utils.h21 namespace tensorforest {
Dparams.cc21 namespace tensorforest { namespace
Dcandidate_graph_runner.h28 namespace tensorforest {
Dgraph_collection_operator.h31 namespace tensorforest {
Dfertile-stats-resource.h32 namespace tensorforest {
Ddecision-tree-resource.h27 namespace tensorforest {
Dtest_utils.h21 namespace tensorforest {
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
Dstochastic_hard_routing_gradient_op.cc40 using tensorforest::LeftProbability;
161 tensorforest::Initialize(*output_routing, 0.0); in Compute()
162 tensorforest::Initialize(*output_data, 0.0); in Compute()
163 tensorforest::Initialize(*output_parameters, 0.0); in Compute()
164 tensorforest::Initialize(*output_bias, 0.0); in Compute()
Dk_feature_routing_function_op.cc43 using tensorforest::CheckTensorBounds;
44 using tensorforest::LeftProbabilityK;
140 tensorforest::GetFeatureSet(layer_num_, i, random_seed_, num_features, in Compute()
Dk_feature_gradient_op.cc36 using tensorforest::LeftProbabilityK;
134 tensorforest::Initialize(*out_data, 0.0f); in Compute()
153 tensorforest::GetFeatureSet(layer_num_, j, random_seed_, num_features, in Compute()
Drouting_function_op.cc43 using tensorforest::CheckTensorBounds;
44 using tensorforest::LeftProbability;
Dhard_routing_function_op.cc43 using tensorforest::CheckTensorBounds;
44 using tensorforest::LeftProbability;
Dstochastic_hard_routing_function_op.cc47 using tensorforest::CheckTensorBounds;
48 using tensorforest::LeftProbability;
Dutils.h24 namespace tensorforest {
Dutils.cc24 namespace tensorforest { namespace
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils.h32 namespace tensorforest {
70 int32 input_feature, const tensorforest::TensorForestDataSpec& spec);
72 int32 input_feature, const tensorforest::TensorForestDataSpec& spec);
172 float bias, const tensorforest::TensorForestDataSpec& spec);
Dtree_utils.cc22 namespace tensorforest { namespace
27 int32 input_feature, const tensorforest::TensorForestDataSpec& spec) { in FindDenseFeatureSpec()
32 int32 input_feature, const tensorforest::TensorForestDataSpec& spec) { in FindSparseFeatureSpec()
563 return tensorforest::FindSparseValue(indices, values, i, feature); in GetSparseFunctor()
575 float bias, const tensorforest::TensorForestDataSpec& spec) { in DecideNode()
/external/tensorflow/tensorflow/contrib/tensor_forest/ops/
Dstats_ops.cc23 namespace tensorforest { namespace

123