Searched refs:num_classes (Results 1 – 6 of 6) sorted by relevance
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/kernels/ |
| A D | ctc_decoder.h | 43 CTCDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCDecoder() argument
44 : num_classes_(num_classes), in CTCDecoder()
45 blank_index_(num_classes - 1), in CTCDecoder()
61 int num_classes() { return num_classes_; } in num_classes() function
74 CTCGreedyDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCGreedyDecoder() argument
75 : CTCDecoder(num_classes, batch_size, merge_repeated) {} in CTCGreedyDecoder()
|
| A D | ctc_beam_search_decoder.cc | 189 const int num_classes = SizeOfDimension(inputs, 2); in Eval() local
207 GetTensorData<float>(inputs) + t * batch_size * num_classes, batch_size, in Eval()
208 num_classes); in Eval()
214 num_classes, beam_width, &beam_scorer, 1 /* batch_size */, in Eval()
219 static_cast<float*>(malloc(num_classes * sizeof(float))); in Eval()
221 dims[0] = num_classes; in Eval()
239 Eigen::Map<const Eigen::ArrayXf>(input_chip_t.data(), num_classes); in Eval()
|
| A D | ctc_beam_search.h | 87 CTCBeamSearchDecoder(int num_classes, int beam_width,
90 : CTCDecoder(num_classes, batch_size, merge_repeated), in CTCDecoder() argument
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/ |
| A D | detection_postprocess.cc | 96 int num_classes; member
143 op_data->num_classes = m["num_classes"].AsInt32(); in Init()
171 const int num_classes = op_data->num_classes; in Prepare() local
502 const int num_classes = op_data->num_classes; in NonMaxSuppressionMultiClassRegularHelper() local
508 int label_offset = num_classes_with_background - num_classes; in NonMaxSuppressionMultiClassRegularHelper()
526 for (int col = 0; col < num_classes; col++) { in NonMaxSuppressionMultiClassRegularHelper()
636 const int num_classes = op_data->num_classes; in NonMaxSuppressionMultiClassFastHelper() local
642 int label_offset = num_classes_with_background - num_classes; in NonMaxSuppressionMultiClassFastHelper()
645 std::min(max_categories_per_anchor, num_classes); in NonMaxSuppressionMultiClassFastHelper()
677 sorted_class_indices + selected_index * num_classes; in NonMaxSuppressionMultiClassFastHelper()
[all …]
|
| /AliOS-Things-master/components/SDL2/src/video/x11/ |
| A D | SDL_x11xinput2.c | 272 for (j = 0; j < dev->num_classes; j++) { in X11_InitXinput2Multitouch()
|
| /AliOS-Things-master/solutions/tflite_micro_speech_demo/micro_speech/train/speech_commands/ |
| A D | train.py | 185 num_classes=label_count)
|
Completed in 9 milliseconds