Searched refs:logits_dimension_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | prediction_ops.cc | 64 context->GetAttr("logits_dimension", &logits_dimension_)); in BoostedTreesTrainingPredictOp()
96 {batch_size, logits_dimension_}, in Compute()
128 std::vector<float> partial_tree_logits(logits_dimension_, 0.0); in Compute()
141 DCHECK_EQ(node_logits.size(), logits_dimension_); in Compute()
142 for (int32 j = 0; j < logits_dimension_; ++j) { in Compute()
150 std::vector<float> partial_all_logits(logits_dimension_, 0.0); in Compute()
154 DCHECK_EQ(leaf_logits.size(), logits_dimension_); in Compute()
157 for (int32 j = 0; j < logits_dimension_; ++j) { in Compute()
175 for (int32 j = 0; j < logits_dimension_; ++j) { in Compute()
192 int32 logits_dimension_; // the size of the output prediction vector. member in tensorflow::BoostedTreesTrainingPredictOp
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/tensor_forest/ |
| D | prediction_ops.cc | 29 context->GetAttr("logits_dimension", &logits_dimension_)); in TensorForestTreePredictOp()
46 context->allocate_output(0, {batch_size, logits_dimension_}, in Compute()
80 for (int j = 0; j < logits_dimension_; ++j) { in set_output_value()
87 int32 logits_dimension_; member in tensorflow::TensorForestTreePredictOp
|