Searched refs:num_classes_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_decoder.h | 44 : num_classes_(num_classes), in CTCDecoder()
61 int num_classes() { return num_classes_; } in num_classes()
64 int num_classes_;
|
| D | ctc_beam_search.h | 220 TFLITE_DCHECK_EQ(num_classes_, input.size()); in GetTopK()
225 for (int j = 0; j < num_classes_ - 1; ++j) { in GetTopK()
239 return std::max((*top_k_logits)[0], input(num_classes_ - 1)); in GetTopK()
251 const int max_classes = top_k ? label_selection_size_ : (num_classes_ - 1); in Step()
275 TFLITE_DCHECK_EQ(num_classes_, raw_input.size()); in Step()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_decoder.h | 44 : num_classes_(num_classes), in CTCDecoder()
61 int num_classes() { return num_classes_; } in num_classes()
64 int num_classes_;
|
| D | ctc_beam_search.h | 223 CHECK_EQ(num_classes_, input.size()); in GetTopK()
228 for (int j = 0; j < num_classes_ - 1; ++j) { in GetTopK()
242 return std::max((*top_k_logits)[0], input(num_classes_ - 1)); in GetTopK()
254 const int max_classes = top_k ? label_selection_size_ : (num_classes_ - 1); in Step()
278 CHECK_EQ(num_classes_, raw_input.size()); in Step()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | prediction_ops.cc | 111 num_classes_ = learner_config.num_classes(); in GradientTreesPredictionOp()
112 OP_REQUIRES(context, num_classes_ >= 2, in GradientTreesPredictionOp()
120 prediction_vector_size_ = reduce_dim ? num_classes_ - 1 : num_classes_; in GradientTreesPredictionOp()
333 int num_classes_; member in tensorflow::boosted_trees::GradientTreesPredictionOp
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | grow_stats.h | 427 : n_(n), num_classes_(num_classes), smallest_weight_class_(-1) {} in FixedSizeClassStats()
449 int num_classes_; variable
|
| D | grow_stats.cc | 675 float w = (smallest_weight / 2.0) * n_ / static_cast<float>(num_classes_); in get_weight()
700 w = (smallest_weight / 2.0) * n_ / static_cast<float>(num_classes_); in set_sum_and_square()
701 *square += (num_classes_ - n_) * w * w; in set_sum_and_square()
|