Home
last modified time | relevance | path

Searched refs:split_types (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dstats_ops_test.py160 right_node_contribs, split_types) = self.evaluate(
179 self.assertAllEqual([_INEQUALITY_DEFAULT_LEFT] * 2, split_types)
222 split_types=['INVALID'] * len(candidate_feature_ids),
238 left_node_contribs, right_node_contribs, split_types) = self.evaluate(
242 split_types=['inequality'] * len(candidate_feature_ids),
262 self.assertAllEqual([_INEQUALITY_DEFAULT_LEFT] * 2, split_types)
275 left_node_contribs, right_node_contribs, split_types) = self.evaluate(
278 split_types=['inequality'],
297 self.assertAllEqual([_INEQUALITY_DEFAULT_LEFT] * 2, split_types)
310 left_node_contribs, right_node_contribs, split_types) = self.evaluate(
[all …]
Dtraining_ops_test.py194 split_types=[
303 split_types=[
416 split_types=[
834 split_types=[
1037 split_types=[
1252 split_types=[
1620 split_types=[group1_split_types])
1777 split_types=[group1_split_types])
1949 split_types=[group1_split_types],
2325 split_types=[
[all …]
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DBoostedTreesUpdateEnsembleV2.pbtxt42 name: "split_types"
106 name: "split_types"
178 name: "split_types"
DBoostedTreesCalculateBestFeatureSplitV2.pbtxt13 name: "split_types"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DBoostedTreesCalculateBestFeatureSplitV2.pbtxt13 name: "split_types"
DBoostedTreesUpdateEnsembleV2.pbtxt43 name: "split_types"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BoostedTreesUpdateEnsembleV2.pbtxt61 name: "split_types"
Dapi_def_BoostedTreesCalculateBestFeatureSplitV2.pbtxt18 name: "split_types"
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc408 const auto& split_types = split_types_list[group_idx].vec<tstring>(); in FindBestSplitsPerNode() local
424 candidate.split_type = split_types(candidate_idx); in FindBestSplitsPerNode()
Dstats_ops.cc642 const auto split_types = split_types_t->vec<tstring>(); in Compute() local
643 DCHECK_EQ(split_types.size(), num_features_); in Compute()
646 if (!(split_types(i) == kInequalitySplit || in Compute()
647 split_types(i) == kEqualitySplit)) { in Compute()
719 const string split_type = split_types(f_idx); in Compute()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt553 …argspec: "args=[\'node_id_range\', \'stats_summaries_list\', \'split_types\', \'candidate_feature_…
645 …\', \'thresholds\', \'left_node_contribs\', \'right_node_contribs\', \'split_types\', \'max_depth\…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt553 …argspec: "args=[\'node_id_range\', \'stats_summaries_list\', \'split_types\', \'candidate_feature_…
645 …\', \'thresholds\', \'left_node_contribs\', \'right_node_contribs\', \'split_types\', \'max_depth\…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go32712 …tput, left_node_contribs []tf.Output, right_node_contribs []tf.Output, split_types []tf.Output, ma…
32723 …left_node_contribs), tf.OutputList(right_node_contribs), tf.OutputList(split_types), max_depth, le…
38995 …ope *Scope, node_id_range tf.Output, stats_summaries_list []tf.Output, split_types tf.Output, cand…
39003 …node_id_range, tf.OutputList(stats_summaries_list), split_types, candidate_feature_ids, l1, l2, tr…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt5707 name: "split_types"
6497 name: "split_types"