/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
H A D | random_tree_gen.h | 20 #include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h" // NOLINT 25 namespace boosted_trees { namespace in namespace:tensorflow 38 boosted_trees::trees::DecisionTreeConfig Generate(int depth); 42 boosted_trees::trees::DecisionTreeConfig Generate( 43 const boosted_trees::trees::DecisionTreeConfig& tree); 46 boosted_trees::trees::DecisionTreeEnsembleConfig GenerateEnsemble( 55 void Combine(boosted_trees::trees::DecisionTreeConfig* root, 56 boosted_trees::trees::DecisionTreeConfig* left_branch, 57 boosted_trees::trees::DecisionTreeConfig* right_branch); 61 void ShiftNodeIndex(boosted_trees [all...] |
H A D | batch_features_testutil.h | 18 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h" 23 namespace boosted_trees { namespace in namespace:tensorflow 39 boosted_trees::utils::BatchFeatures* batch_features); 42 } // namespace boosted_trees
|
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
H A D | split_handler_ops.py | 21 from tensorflow.contrib.boosted_trees.python.ops import boosted_trees_ops_loader 24 from tensorflow.contrib.boosted_trees.python.ops.gen_split_handler_ops import *
|
H A D | training_ops.py | 21 from tensorflow.contrib.boosted_trees.python.ops import boosted_trees_ops_loader 24 from tensorflow.contrib.boosted_trees.python.ops.gen_training_ops import *
|
H A D | prediction_ops.py | 21 from tensorflow.contrib.boosted_trees.python.ops import boosted_trees_ops_loader 22 from tensorflow.contrib.boosted_trees.python.ops.gen_prediction_ops import gradient_trees_partition_examples 23 from tensorflow.contrib.boosted_trees.python.ops.gen_prediction_ops import gradient_trees_prediction
|
H A D | model_ops.py | 21 from tensorflow.contrib.boosted_trees.python.ops import boosted_trees_ops_loader 23 from tensorflow.contrib.boosted_trees.python.ops import gen_model_ops 24 from tensorflow.contrib.boosted_trees.python.ops.gen_model_ops import tree_ensemble_deserialize 25 from tensorflow.contrib.boosted_trees.python.ops.gen_model_ops import tree_ensemble_serialize 27 from tensorflow.contrib.boosted_trees.python.ops.gen_model_ops import tree_ensemble_stamp_token
|
/external/tensorflow/tensorflow/contrib/boosted_trees/ |
H A D | __init__.py | 21 from tensorflow.contrib.boosted_trees.python import *
|
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
H A D | __init__.py | 21 from tensorflow.contrib.boosted_trees.estimator_batch import *
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/ |
H A D | multiple_additive_trees.h | 20 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h" 21 #include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h" // NOLINT 27 namespace boosted_trees { namespace in namespace:tensorflow 37 const boosted_trees::trees::DecisionTreeEnsembleConfig& config, 39 const boosted_trees::utils::BatchFeatures& features, 45 } // namespace boosted_trees
|
H A D | multiple_additive_trees.cc | 15 #include "tensorflow/contrib/boosted_trees/lib/models/multiple_additive_trees.h" 16 #include "tensorflow/contrib/boosted_trees/lib/trees/decision_tree.h" 17 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h" 18 #include "tensorflow/contrib/boosted_trees/lib/utils/parallel_for.h" 21 namespace boosted_trees { namespace in namespace:tensorflow 25 const boosted_trees::trees::DecisionTreeEnsembleConfig& config, 27 const boosted_trees::utils::BatchFeatures& features, 45 const boosted_trees::trees::DecisionTreeConfig& tree = 76 boosted_trees::utils::ParallelFor(batch_size, worker_threads->NumThreads(), 81 } // namespace boosted_trees [all...] |
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ |
H A D | __init__.py | 21 from tensorflow.contrib.boosted_trees.python.ops import *
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/ |
H A D | example_partitioner.cc | 15 #include "tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/example_partitioner.h" 16 #include "tensorflow/contrib/boosted_trees/lib/utils/parallel_for.h" 19 namespace boosted_trees { namespace in namespace:tensorflow 23 const boosted_trees::trees::DecisionTreeConfig& tree_config, 24 const boosted_trees::utils::BatchFeatures& features, 41 example_partition = boosted_trees::trees::DecisionTree::Traverse( 51 boosted_trees::utils::ParallelFor(batch_size, desired_parallelism, 56 const boosted_trees::trees::DecisionTreeConfig& tree_config, 57 const boosted_trees::utils::BatchFeatures& features, 73 uint32 partition = boosted_trees [all...] |
H A D | example_partitioner_test.cc | 15 #include "tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/example_partitioner.h" 21 namespace boosted_trees { namespace in namespace:tensorflow 37 boosted_trees::utils::BatchFeatures batch_features_; 41 boosted_trees::trees::DecisionTreeConfig tree_config; 53 boosted_trees::trees::DecisionTreeConfig tree_config; 76 boosted_trees::trees::DecisionTreeConfig tree_config; 98 } // namespace boosted_trees
|
H A D | example_partitioner.h | 19 #include "tensorflow/contrib/boosted_trees/lib/trees/decision_tree.h" 20 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h" 24 namespace boosted_trees { namespace in namespace:tensorflow 50 } // namespace boosted_trees
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/stats/ |
H A D | feature-split-candidate.h | 19 #include "tensorflow/contrib/boosted_trees/lib/learner/common/stats/split-stats.h" 20 #include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h" 23 namespace boosted_trees { namespace in namespace:tensorflow 36 const boosted_trees::trees::TreeNode& node, 47 boosted_trees::trees::TreeNode tree_node; 58 } // namespace boosted_trees
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/quantiles/ |
H A D | weighted_quantiles_buffer_test.cc | 15 #include "tensorflow/contrib/boosted_trees/lib/quantiles/weighted_quantiles_buffer.h" 26 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double>; 28 boosted_trees::quantiles::WeightedQuantilesBuffer<double, 36 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double> 42 boosted_trees::quantiles::WeightedQuantilesBuffer<double, double>
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/trees/ |
H A D | decision_tree.h | 18 #include "tensorflow/contrib/boosted_trees/lib/utils/example.h" 19 #include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h" // NOLINT 22 namespace boosted_trees { namespace in namespace:tensorflow 46 } // namespace boosted_trees
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
H A D | parallel_for.h | 21 namespace boosted_trees { namespace in namespace:tensorflow 30 } // namespace boosted_trees
|
H A D | random.h | 21 namespace boosted_trees { namespace in namespace:tensorflow 36 } // namespace boosted_trees
|
H A D | optional_value.h | 22 namespace boosted_trees { namespace in namespace:tensorflow 44 } // namespace boosted_trees
|
H A D | dropout_utils.h | 21 #include "tensorflow/contrib/boosted_trees/proto/learner.pb.h" // NOLINT 26 namespace boosted_trees { namespace in namespace:tensorflow 74 } // namespace boosted_trees
|
/external/tensorflow/tensorflow/contrib/boosted_trees/resources/ |
H A D | stamped_resource.h | 22 namespace boosted_trees { namespace in namespace:tensorflow 40 } // namespace boosted_trees
|
H A D | decision_tree_ensemble_resource.h | 18 #include "tensorflow/contrib/boosted_trees/lib/trees/decision_tree.h" 19 #include "tensorflow/contrib/boosted_trees/resources/stamped_resource.h" 25 namespace boosted_trees { namespace in namespace:tensorflow 35 boosted_trees::trees::DecisionTreeEnsembleConfig>(&arena_)) {} 42 const boosted_trees::trees::DecisionTreeEnsembleConfig& 65 boosted_trees::trees::GrowingMetadata* const growing_metadata = 76 boosted_trees::trees::DecisionTreeConfig* AddNewTree(const float weight) { 79 boosted_trees::trees::DecisionTreeMetadata* const metadata = 92 boosted_trees::trees::DecisionTreeConfig* LastTree() { 98 boosted_trees [all...] |
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
H A D | model_ops.cc | 17 #include "tensorflow/contrib/boosted_trees/lib/utils/tensor_utils.h" 18 #include "tensorflow/contrib/boosted_trees/resources/decision_tree_ensemble_resource.h" 27 namespace boosted_trees { namespace in namespace:tensorflow 29 using boosted_trees::models::DecisionTreeEnsembleResource; 47 auto* result = new boosted_trees::models::DecisionTreeEnsembleResource(); 72 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; 91 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; 115 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; 144 IsResourceInitialized<boosted_trees::models::DecisionTreeEnsembleResource>); 158 } // namespace boosted_trees [all...] |
H A D | training_ops.cc | 15 #include "tensorflow/contrib/boosted_trees/lib/utils/dropout_utils.h" 16 #include "tensorflow/contrib/boosted_trees/proto/learner.pb.h" 17 #include "tensorflow/contrib/boosted_trees/proto/split_info.pb.h" 18 #include "tensorflow/contrib/boosted_trees/resources/decision_tree_ensemble_resource.h" 23 using tensorflow::boosted_trees::learner::LearningRateDropoutDrivenConfig; 25 namespace boosted_trees { namespace in namespace:tensorflow 29 using boosted_trees::learner::LearningRateConfig; 30 using boosted_trees::trees::Leaf; 31 using boosted_trees::trees::TreeNode; 32 using boosted_trees [all...] |