Home
last modified time | relevance | path

Searched defs:output_classes (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Ddetection_post_process_test.cc79 auto *output_classes = new lite::Tensor; in DetectionPostProcessTestInit() local
139 auto *output_classes = reinterpret_cast<float *>(outputs_[1]->MutableData()); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/base/
Ddetection_post_process_base_coder.cc142 Tensor *output_classes = output_tensors_.at(1); in DoCode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/
Ddetection_post_process_base.cc240 auto output_classes = reinterpret_cast<float *>(out_tensors_.at(SECOND_INPUT)->data()); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Ddetection_post_process_fp32.c133 const float *decoded_boxes, float *output_boxes, float *output_classes, in DetectionPostProcessFast()
171 … float *output_boxes, float *output_classes, float *output_scores, float *output_num, in DetectionPostProcessRegular()