| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | prediction_ops_test.py | 58 logits_dimension=1)
87 logits_dimension = 2
95 logits_dimension=logits_dimension)
163 logits_dimension=1)
232 logits_dimension=1)
304 logits_dimension=1)
378 logits_dimension=1)
451 logits_dimension=1)
535 logits_dimension = 2
543 logits_dimension=logits_dimension)
[all …]
|
| D | stats_ops_test.py | 147 logits_dimension=1))
170 logits_dimension=1))
228 logits_dimension=1))
248 logits_dimension=1))
284 logits_dimension=1))
319 logits_dimension=1))
354 logits_dimension=1))
393 logits_dimension=1))
461 logits_dimension=1))
495 logits_dimension=1))
[all …]
|
| D | training_ops_test.py | 374 logits_dimension = 2
419 logits_dimension=logits_dimension)
1196 logits_dimension = 2
1255 logits_dimension=logits_dimension)
1925 logits_dimension = 2
1950 logits_dimension=logits_dimension)
2269 logits_dimension = 2
2328 logits_dimension=logits_dimension)
3011 logits_dimension = 2
3050 logits_dimension=logits_dimension)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | resources.cc | 257 const int32 logits_dimension) { in AddNewTree() argument
258 const std::vector<float> empty_leaf(logits_dimension); in AddNewTree()
259 return AddNewTreeWithLogits(weight, empty_leaf, logits_dimension); in AddNewTree()
264 const int32 logits_dimension) { in AddNewTreeWithLogits() argument
267 if (logits_dimension == 1) { in AddNewTreeWithLogits()
270 for (int32 i = 0; i < logits_dimension; ++i) { in AddNewTreeWithLogits()
283 const int32 logits_dimension, int32* left_node_id, int32* right_node_id) { in AddBucketizedSplitNode() argument
285 auto* node = AddLeafNodes(tree_id, split_entry, logits_dimension, in AddBucketizedSplitNode()
308 const int32 logits_dimension, int32* left_node_id, int32* right_node_id) { in AddCategoricalSplitNode() argument
310 auto* node = AddLeafNodes(tree_id, split_entry, logits_dimension, in AddCategoricalSplitNode()
[all …]
|
| D | resources.h | 105 int32 AddNewTree(const float weight, const int32 logits_dimension);
110 const int32 logits_dimension);
116 const int32 logits_dimension, int32* left_node_id, int32* right_node_id);
122 const int32 logits_dimension, int32* left_node_id, int32* right_node_id);
144 void PostPruneTree(const int32 current_tree, const int32 logits_dimension);
182 const int32 logits_dimension, int32* left_node_id, int32* right_node_id);
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tensor_forest_ops.cc | 74 int logits_dimension; in __anon6270b1470302() local
75 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anon6270b1470302()
76 c->set_output(0, c->Matrix(batch_size, logits_dimension)); in __anon6270b1470302()
|
| D | boosted_trees_ops.cc | 135 int logits_dimension; in __anon770f674c0302() local
136 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anon770f674c0302()
138 c->MakeShape({c->UnknownDim(), logits_dimension}); in __anon770f674c0302()
167 int logits_dimension; in __anon770f674c0402() local
168 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anon770f674c0402()
205 c->MakeShape({c->UnknownDim(), logits_dimension}); in __anon770f674c0402()
253 int logits_dimension; in __anon770f674c0502() local
254 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anon770f674c0502()
256 c->MakeShape({c->UnknownDim(), logits_dimension}); in __anon770f674c0502()
475 int logits_dimension; in __anon770f674c0c02() local
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesCenterBias.pbtxt | 13 A tensor with shape=[logits_dimension] with mean of gradients for a first node.
19 A tensor with shape=[logits_dimension] mean of hessians for a first node.
|
| D | api_def_BoostedTreesAggregateStats.pbtxt | 13 float32; Rank 2 Tensor (shape=[batch_size, logits_dimension]) with gradients for each example.
31 output Rank 4 Tensor (shape=[splits, feature_dimension, buckets, logits_dimension + hessian_dimensi…
|
| D | api_def_BoostedTreesSparseAggregateStats.pbtxt | 13 float32; Rank 2 Tensor (shape=[batch_size, logits_dimension]) with gradients for each example.
51 statistics_dimension = logits_dimension + hessian_dimension.
|
| D | api_def_TensorForestTreePredict.pbtxt | 5 name: "logits_dimension"
|
| D | api_def_BoostedTreesPredict.pbtxt | 24 name: "logits_dimension"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BoostedTreesCalculateBestFeatureSplit.pbtxt | 56 name: "logits_dimension"
129 name: "logits_dimension"
|
| D | BoostedTreesUpdateEnsembleV2.pbtxt | 128 name: "logits_dimension"
200 name: "logits_dimension"
|
| D | TensorForestTreePredict.pbtxt | 16 name: "logits_dimension"
|
| D | BoostedTreesExampleDebugOutputs.pbtxt | 23 name: "logits_dimension"
|
| D | BoostedTreesPredict.pbtxt | 23 name: "logits_dimension"
|
| D | BoostedTreesTrainingPredict.pbtxt | 39 name: "logits_dimension"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TensorForestTreePredict.pbtxt | 16 name: "logits_dimension"
|
| D | BoostedTreesExampleDebugOutputs.pbtxt | 23 name: "logits_dimension"
|
| D | BoostedTreesPredict.pbtxt | 23 name: "logits_dimension"
|
| D | BoostedTreesTrainingPredict.pbtxt | 39 name: "logits_dimension"
|
| D | BoostedTreesCalculateBestFeatureSplit.pbtxt | 56 name: "logits_dimension"
|
| D | BoostedTreesCalculateBestFeatureSplitV2.pbtxt | 75 name: "logits_dimension"
|
| D | BoostedTreesSparseCalculateBestFeatureSplit.pbtxt | 64 name: "logits_dimension"
|