Searched refs:gains_list (Results 1 – 9 of 9) 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,
70 self.evaluate(gains_list))
105 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
118 self.evaluate(gains_list))
154 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
176 self.evaluate(gains_list))
206 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
220 self.evaluate(gains_list))
256 (node_ids_list, gains_list, thresholds_list, left_node_contribs_list,
270 self.assertAllClose([[0.384314], [0.098013]], self.evaluate(gains_list))
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | training_ops.cc | 67 OpInputList gains_list; in Compute() local
72 OP_REQUIRES_OK(context, context->input_list("gains", &gains_list)); in Compute()
94 FindBestSplitsPerNode(context, node_ids_list, gains_list, feature_ids, in Compute()
124 const float gain = gains_list[feature_idx].vec<float>()(candidate_idx); in Compute()
198 const OpInputList& gains_list, in FindBestSplitsPerNode() argument
204 const auto& gains = gains_list[feature_idx].vec<float>(); in FindBestSplitsPerNode()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | training_ops.cc | 366 OpInputList gains_list; in Compute() local
370 OP_REQUIRES_OK(context, context->input_list("gains", &gains_list)); in Compute()
381 gains_list, splits_list, &best_splits); in Compute()
385 FindBestSplitOblivious(context, gains_list, splits_list, &best_splits); in Compute()
443 const OpInputList& gains_list, const OpInputList& splits_list, in FindBestSplitsPerPartitionNormal() argument
449 const auto& gains = gains_list[handler_id].vec<float>(); in FindBestSplitsPerPartitionNormal()
478 const OpInputList& gains_list, in FindBestSplitOblivious() argument
483 const auto& gains = gains_list[handler_id].vec<float>(); in FindBestSplitOblivious()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 1057 gains_list = []
1066 gains_list.append(gains)
1074 gains = array_ops.concat(gains_list, axis=0)
1106 gains_list = list(array_ops.split(gains, sizes, axis=0))
1114 gains=gains_list,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt | 47 name: "gains_list"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 31367 …n_node_weight tf.Output, max_splits int64) (node_ids_list []tf.Output, gains_list []tf.Output, thr…
31389 if gains_list, idx, err = makeOutputList(op, idx, "gains_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 | 4423 name: "gains_list"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 11637 name: "gains_list"
|
| D | ops_history.v2.pbtxt | 11637 name: "gains_list"
|