Searched refs:_class_scores (Results 1 – 2 of 2) sorted by relevance
214 _selected_indices(), _class_scores(), _input_scores_to_use(nullptr) in CPPDetectionPostProcessLayer()254 …auto_init_if_empty(*_class_scores.info(), TensorInfo(info.use_regular_nms() ? TensorShape(_num_box… in configure()262 _memory_group.manage(&_class_scores); in configure()263 …_nms.configure(&_decoded_boxes, &_class_scores, &_selected_indices, info.use_regular_nms() ? info.… in configure()269 _class_scores.allocator()->allocate(); in configure()335 *(reinterpret_cast<float *>(_class_scores.ptr_to_element(Coordinates(i)))) = in run()351 …result_scores_after_nms.emplace_back((reinterpret_cast<float *>(_class_scores.buffer()))[selected_… in run()405 …*(reinterpret_cast<float *>(_class_scores.ptr_to_element(Coordinates(b * num_classes_per_box + i))… in run()
111 Tensor _class_scores; variable