Searched refs:thresholds_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,
71 self.assertAllEqual([[1, 1], [1, 1]], self.evaluate(thresholds_list))
105 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
119 self.assertAllEqual([[0, 1], [1, 1]], self.evaluate(thresholds_list))
154 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
165 self.assertAllEqual([[0, 1], [1, 1]], self.evaluate(thresholds_list))
206 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
222 self.assertAllEqual([[0, 1], [1, 1]], self.evaluate(thresholds_list))
256 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
271 self.assertAllEqual([[1], [1]], self.evaluate(thresholds_list))
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | training_ops.cc | 68 OpInputList thresholds_list; in Compute() local
74 context->input_list("thresholds", &thresholds_list)); in Compute()
123 thresholds_list[feature_idx].vec<int32>()(candidate_idx); in Compute()
|
| /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 | 31367 …ax_splits int64) (node_ids_list []tf.Output, gains_list []tf.Output, thresholds_list []tf.Output, …
31393 if thresholds_list, idx, err = makeOutputList(op, idx, "thresholds_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 | 4428 name: "thresholds_list"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 11642 name: "thresholds_list"
|
| D | ops_history.v2.pbtxt | 11642 name: "thresholds_list"
|