| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/trees/ |
| D | decision_tree_test.cc | 95 DecisionTreeConfig tree_config; in TEST_F() local
97 EXPECT_EQ(-1, DecisionTree::Traverse(tree_config, 0, example)); in TEST_F()
101 DecisionTreeConfig tree_config; in TEST_F() local
102 tree_config.add_nodes()->mutable_leaf(); in TEST_F()
104 EXPECT_EQ(0, DecisionTree::Traverse(tree_config, 0, example)); in TEST_F()
108 DecisionTreeConfig tree_config; in TEST_F() local
109 tree_config.add_nodes()->mutable_leaf(); in TEST_F()
111 EXPECT_EQ(-1, DecisionTree::Traverse(tree_config, 10, example)); in TEST_F()
115 DecisionTreeConfig tree_config; in TEST_F() local
117 tree_config.add_nodes()->mutable_dense_float_binary_split(); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/ |
| D | example_partitioner_test.cc | 41 boosted_trees::trees::DecisionTreeConfig tree_config; in TEST_F() local
43 ExamplePartitioner::UpdatePartitions(tree_config, batch_features_, 1, in TEST_F()
53 boosted_trees::trees::DecisionTreeConfig tree_config; in TEST_F() local
54 auto* split = tree_config.add_nodes()->mutable_dense_float_binary_split(); in TEST_F()
59 tree_config.add_nodes()->mutable_leaf(); in TEST_F()
60 tree_config.add_nodes()->mutable_leaf(); in TEST_F()
66 ExamplePartitioner::UpdatePartitions(tree_config, batch_features_, 1, in TEST_F()
76 boosted_trees::trees::DecisionTreeConfig tree_config; in TEST_F() local
77 auto* split = tree_config.add_nodes()->mutable_dense_float_binary_split(); in TEST_F()
82 tree_config.add_nodes()->mutable_leaf(); in TEST_F()
[all …]
|
| D | example_partitioner.cc | 23 const boosted_trees::trees::DecisionTreeConfig& tree_config, in UpdatePartitions() argument
34 auto partition_examples_subset = [&tree_config, &features, in UpdatePartitions()
37 if (TF_PREDICT_TRUE(tree_config.nodes_size() > 0)) { in UpdatePartitions()
42 tree_config, example_partition, example); in UpdatePartitions()
56 const boosted_trees::trees::DecisionTreeConfig& tree_config, in PartitionExamples() argument
67 auto partition_examples_subset = [&tree_config, &features, in PartitionExamples()
70 if (TF_PREDICT_TRUE(tree_config.nodes_size() > 0)) { in PartitionExamples()
74 tree_config, 0, example); in PartitionExamples()
|
| D | example_partitioner.h | 33 static void UpdatePartitions(const trees::DecisionTreeConfig& tree_config,
42 static void PartitionExamples(const trees::DecisionTreeConfig& tree_config,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | training_ops.cc | 265 boosted_trees::trees::DecisionTreeConfig* const tree_config = in RetrieveBias() local
267 auto* const leaf = tree_config->add_nodes()->mutable_leaf(); in RetrieveBias()
274 boosted_trees::trees::DecisionTreeConfig* const tree_config = in RetrieveBias() local
276 CHECK_EQ(tree_config->nodes_size(), 1); in RetrieveBias()
277 CHECK_EQ(tree_config->nodes(0).node_case(), TreeNode::kLeaf); in RetrieveBias()
278 return tree_config->mutable_nodes(0)->mutable_leaf(); in RetrieveBias()
403 boosted_trees::trees::DecisionTreeConfig* const tree_config = in Compute() local
411 SplitTreeNode(split_entry.first, &split_entry.second, tree_config, in Compute()
417 SplitTreeLayer(&best_splits[0], tree_config, ensemble_resource); in Compute()
428 PruneTree(tree_config); in Compute()
[all …]
|
| D | prediction_ops.cc | 396 const boosted_trees::trees::DecisionTreeConfig& tree_config = in DoCompute() local
455 tree_config, batch_features, worker_threads->NumThreads(), in DoCompute()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_forest_ops.py | 76 def tree_variable(tree_config, name, container=None): argument
78 'TreeVariable', name, container, tree_config,
94 tree_config = ''
96 tree_config = tree_configs[i]
98 tree_config,
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/ |
| D | model_ops.py | 98 def __init__(self, params, tree_config, stats_handle, name, container=None): argument
100 self._tree_config = tree_config
148 def tree_variable(params, tree_config, stats_handle, name, container=None): argument
162 tree_var = TreeVariable(params, tree_config, stats_handle, name, container)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TensorForestTreeDeserialize.pbtxt | 11 name: "tree_config"
|
| D | api_def_TensorForestTreeSerialize.pbtxt | 11 name: "tree_config"
|
| D | api_def_TensorForestCreateTreeVariable.pbtxt | 11 name: "tree_config"
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/proto/ |
| D | BUILD | 37 srcs = ["tree_config.proto"],
|
| D | split_info.proto | 7 import "tensorflow/contrib/boosted_trees/proto/tree_config.proto";
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ |
| D | tensor_forest.py | 307 def __init__(self, params, tree_num, training, tree_config='', tree_stat=''): argument
321 params, tree_config, self.stats, self.get_tree_name('tree', tree_num))
354 kwargs.update(dict(tree_config=tree_configs[i]))
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_proto_files.txt | 4 tensorflow/contrib/boosted_trees/proto/tree_config.proto
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 3698 func TensorForestCreateTreeVariable(scope *Scope, tree_handle tf.Output, tree_config tf.Output) (o …
3705 tree_handle, tree_config,
7609 func TensorForestTreeDeserialize(scope *Scope, tree_handle tf.Output, tree_config tf.Output) (o *tf…
7616 tree_handle, tree_config,
14593 func TensorForestTreeSerialize(scope *Scope, tree_handle tf.Output) (tree_config tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 41274 name: "tree_config"
41286 name: "tree_config"
41352 name: "tree_config"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 84126 name: "tree_config"
84138 name: "tree_config"
84204 name: "tree_config"
|
| D | ops_history.v1.pbtxt | 84753 name: "tree_config"
84765 name: "tree_config"
84831 name: "tree_config"
|