Searched refs:logits_dimension_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | prediction_ops.cc | 50 context->GetAttr("logits_dimension", &logits_dimension_)); in BoostedTreesTrainingPredictOp()
51 OP_REQUIRES(context, logits_dimension_ == 1, in BoostedTreesTrainingPredictOp()
89 {batch_size, logits_dimension_}, in Compute()
177 int32 logits_dimension_; // the size of the output prediction vector. member in tensorflow::BoostedTreesTrainingPredictOp
192 context->GetAttr("logits_dimension", &logits_dimension_)); in BoostedTreesPredictOp()
217 "logits", {batch_size, logits_dimension_}, in Compute()
231 std::vector<float> tree_logits(logits_dimension_, 0.0); in Compute()
238 DCHECK_EQ(leaf_logits.size(), logits_dimension_); in Compute()
239 for (int32 j = 0; j < logits_dimension_; ++j) { in Compute()
254 for (int32 j = 0; j < logits_dimension_; ++j) { in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/tensor_forest/ |
| D | prediction_ops.cc | 28 context->GetAttr("logits_dimension", &logits_dimension_)); in TensorForestTreePredictOp()
47 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
|