Home
last modified time | relevance | path

Searched refs:split_node (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/trees/
Ddecision_tree_test.cc116 auto* split_node = in TEST_F() local
118 split_node->set_feature_column(0); in TEST_F()
119 split_node->set_threshold(0.0f); in TEST_F()
120 split_node->set_left_id(1); in TEST_F()
121 split_node->set_right_id(2); in TEST_F()
140 auto* split_node = tree_config.add_nodes() in TEST_F() local
143 split_node->set_feature_column(0); in TEST_F()
144 split_node->set_threshold(-20.0f); in TEST_F()
145 split_node->set_left_id(1); in TEST_F()
146 split_node->set_right_id(2); in TEST_F()
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
Dsplit_handler_ops_test.py87 split_node = split_info.split_node.dense_float_binary_split
93 self.assertEqual(0, split_node.feature_column)
94 self.assertAllClose(0.3, split_node.threshold, 0.00001)
104 split_node = split_info.split_node.dense_float_binary_split
109 self.assertEqual(0, split_node.feature_column)
110 self.assertAllClose(0.52, split_node.threshold, 0.00001)
146 split_node = split_info.split_node.dense_float_binary_split
151 self.assertEqual(0, split_node.feature_column)
152 self.assertAllClose(0.3, split_node.threshold, 1e-6)
238 split_node = split_info.split_node.sparse_float_binary_split_default_right
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/
Dordinal_split_handler_test.py145 split_node = split_info.split_node.dense_float_binary_split
154 self.assertEqual(0, split_node.feature_column)
156 self.assertAllClose(0.3, split_node.threshold, 0.00001)
174 split_node = split_info.split_node.dense_float_binary_split
181 self.assertEqual(0, split_node.feature_column)
183 self.assertAllClose(0.52, split_node.threshold, 0.00001)
266 split_node = oblivious_split_info.split_node
267 split_node = split_node.oblivious_dense_float_binary_split
268 self.assertAllClose(0.3, split_node.threshold, 0.00001)
269 self.assertEqual(0, split_node.feature_column)
[all …]
Dcategorical_split_handler_test.py131 split_node = split_info.split_node.categorical_id_binary_split
133 self.assertEqual(0, split_node.feature_column)
135 self.assertEqual(2, split_node.feature_id)
161 split_node = split_info.split_node.categorical_id_binary_split
168 self.assertEqual(0, split_node.feature_column)
170 self.assertEqual(1, split_node.feature_id)
255 split_node = split_info.split_node.oblivious_categorical_id_binary_split
257 self.assertEqual(0, split_node.feature_column)
258 self.assertEqual(1, split_node.feature_id)
365 split_node = split_info.split_node.categorical_id_binary_split
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/proto/
Dsplit_info.proto12 tensorflow.boosted_trees.trees.TreeNode split_node = 1; field
22 tensorflow.boosted_trees.trees.TreeNode split_node = 1; field
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc66 if (TF_PREDICT_FALSE(split->split_info.split_node().node_case() == in UpdateBestSplit()
100 << "(current split: " << split->split_info.split_node().node_case() in UpdateBestSplit()
102 << best_split.split_info.split_node().node_case() << ")"; in UpdateBestSplit()
103 if (split->split_info.split_node().node_case() < in UpdateBestSplit()
104 best_split.split_info.split_node().node_case()) { in UpdateBestSplit()
106 } else if (split->split_info.split_node().node_case() == in UpdateBestSplit()
107 best_split.split_info.split_node().node_case()) { in UpdateBestSplit()
537 trees::TreeNode current_node = split_info.split_node(); in FindBestSplitOblivious()
538 trees::TreeNode best_node = best_split.oblivious_split_info.split_node(); in FindBestSplitOblivious()
728 CHECK(split->split_info.split_node().node_case() != TreeNode::NODE_NOT_SET); in SplitTreeNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc562 auto split_node = node_map.GetNode("split"); in TEST_F() local
563 EXPECT_EQ(split_node->input(0), "split-0-LayoutOptimizer"); in TEST_F()
564 EXPECT_EQ(split_node->input(1), "Conv2D"); in TEST_F()
582 auto split_node = node_map.GetNode("split"); in TEST_F() local
583 EXPECT_EQ(split_node->input(0), "split-0-LayoutOptimizer"); in TEST_F()
584 EXPECT_EQ(split_node->input(1), "Conv2D"); in TEST_F()
602 auto split_node = node_map.GetNode("split"); in TEST_F() local
603 EXPECT_EQ(split_node->input(0), "split-0-LayoutOptimizer"); in TEST_F()
604 EXPECT_EQ(split_node->input(1), "Conv2D"); in TEST_F()
622 auto split_node = node_map.GetNode("split"); in TEST_F() local
[all …]
Darithmetic_optimizer.cc1653 NodeDef* split_node) { in HoistChainForSplit() argument
1655 const string& split_name = split_node->name(); in HoistChainForSplit()
1666 const int value_slot = split_node->op() == "SplitV" ? 0 : 1; in HoistChainForSplit()
1667 const string orig_input = split_node->input(value_slot); in HoistChainForSplit()
1668 split_node->set_input(value_slot, cur_copy->name()); in HoistChainForSplit()
1669 ctx().node_map->UpdateInput(split_node->name(), orig_input, in HoistChainForSplit()
1674 while (cur_tail != split_node) { in HoistChainForSplit()
/external/e2fsprogs/debugfs/
Dextent_cmds.ct53 split_node, split;
/external/e2fsprogs/doc/RelNotes/
Dv1.42.txt639 extent_inode subcommands which take arguments (split_node,
/external/e2fsprogs/debian/
Dchangelog874 extent_inode subcommands which take arguments (split_node,