Searched refs:node_ids_list (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 57 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
68 self.assertAllEqual([[1, 2], [1, 2]], self.evaluate(node_ids_list))
105 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
116 self.assertAllEqual([[1, 2], [1, 2]], self.evaluate(node_ids_list))
154 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
167 self.assertAllEqual([[1, 2], [1, 2]], self.evaluate(node_ids_list))
206 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
217 self.assertAllEqual([[1, 2], [1, 2]], self.evaluate(node_ids_list))
256 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
269 self.assertAllEqual([[2], [1]], self.evaluate(node_ids_list))
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | training_ops.cc | 66 OpInputList node_ids_list; in Compute() local
71 OP_REQUIRES_OK(context, context->input_list("node_ids", &node_ids_list)); in Compute()
94 FindBestSplitsPerNode(context, node_ids_list, gains_list, feature_ids, in Compute()
197 OpKernelContext* const context, const OpInputList& node_ids_list, in FindBestSplitsPerNode() argument
203 const auto& node_ids = node_ids_list[feature_idx].vec<int32>(); in FindBestSplitsPerNode()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt | 41 name: "node_ids_list"
61 …ranching from parent nodes (given by the tensor element in the output node_ids_list) to the left d…
86 …possible nodes can differ between the features. Therefore, we return `node_ids_list` for each feat…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 31367 …e_complexity tf.Output, min_node_weight tf.Output, max_splits int64) (node_ids_list []tf.Output, g…
31385 if node_ids_list, idx, err = makeOutputList(op, idx, "node_ids_list"); err != nil {
31405 …return node_ids_list, gains_list, thresholds_list, left_node_contribs_list, right_node_contribs_li…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 4418 name: "node_ids_list"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 11632 name: "node_ids_list"
|
| D | ops_history.v2.pbtxt | 11632 name: "node_ids_list"
|