Searched refs:thresholds_list (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | training_ops.cc | 57 OpInputList thresholds_list; in Compute() local
63 context->input_list("thresholds", &thresholds_list)); in Compute()
85 thresholds_list, left_node_contribs, in Compute()
172 const OpInputList& thresholds_list, in FindBestSplitsPerNode() argument
181 const auto& thresholds = thresholds_list[feature_idx].vec<int32>(); in FindBestSplitsPerNode()
253 OpInputList thresholds_list; in Compute() local
261 context->input_list("thresholds", &thresholds_list)); in Compute()
290 thresholds_list, dimension_ids_list, in Compute()
392 const OpInputList& thresholds_list, const OpInputList& dimension_ids_list, in FindBestSplitsPerNode() argument
402 const auto& thresholds = thresholds_list[group_idx].vec<int32>(); in FindBestSplitsPerNode()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BoostedTreesCalculateBestGainsPerFeature.pbtxt | 39 name: "thresholds_list"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BoostedTreesCalculateBestGainsPerFeature.pbtxt | 39 name: "thresholds_list"
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 188 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
202 self.assertAllEqual([[1, 1], [1, 1]], self.evaluate(thresholds_list))
419 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
433 self.assertAllEqual([[0, 1], [1, 1]], self.evaluate(thresholds_list))
582 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
593 self.assertAllEqual([[0, 1], [1, 1]], self.evaluate(thresholds_list))
751 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
767 self.assertAllEqual([[0, 1], [1, 1]], self.evaluate(thresholds_list))
933 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
948 self.assertAllEqual([[1], [1]], self.evaluate(thresholds_list))
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt | 53 name: "thresholds_list"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 3669 …ax_splits int64) (node_ids_list []tf.Output, gains_list []tf.Output, thresholds_list []tf.Output, …
3695 if thresholds_list, idx, err = makeOutputList(op, idx, "thresholds_list"); err != nil {
3707 …return node_ids_list, gains_list, thresholds_list, left_node_contribs_list, right_node_contribs_li…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 5686 name: "thresholds_list"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 11642 name: "thresholds_list"
|