Searched refs:decoded_boxes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | detection_postprocess.cc | 211 TfLiteTensor* decoded_boxes = &context->tensors[op_data->decoded_boxes_index]; in Prepare() local
212 decoded_boxes->type = kTfLiteFloat32; in Prepare()
213 decoded_boxes->allocation_type = kTfLiteArenaRw; in Prepare()
214 SetTensorSizes(context, decoded_boxes, in Prepare()
342 TfLiteTensor* decoded_boxes = in DecodeCenterSizeBoxes() local
344 TF_LITE_ENSURE_EQ(context, decoded_boxes->type, kTfLiteFloat32); in DecodeCenterSizeBoxes()
345 auto& box = ReInterpretTensor<BoxCornerEncoding*>(decoded_boxes)[idx]; in DecodeCenterSizeBoxes()
388 bool ValidateBoxes(const TfLiteTensor* decoded_boxes, const int num_boxes) { in ValidateBoxes() argument
390 auto& box = ReInterpretTensor<const BoxCornerEncoding*>(decoded_boxes)[i]; in ValidateBoxes()
401 float ComputeIntersectionOverUnion(const TfLiteTensor* decoded_boxes, in ComputeIntersectionOverUnion() argument
[all …]
|
| /external/ComputeLibrary/src/runtime/CPP/functions/ |
| D | CPPDetectionPostProcessLayer.cpp | 121 …_encoding, const ITensor *input_anchors, DetectionPostProcessLayerInfo info, Tensor *decoded_boxes) in DecodeCenterSizeBoxes() argument
134 Iterator decoded_it(decoded_boxes, win); in DecodeCenterSizeBoxes()
182 void SaveOutputs(const Tensor *decoded_boxes, const std::vector<int> &result_idx_boxes_after_nms, c… in SaveOutputs() argument
191 …es->ptr_to_element(Coordinates(0, i)))) = *(reinterpret_cast<float *>(decoded_boxes->ptr_to_elemen… in SaveOutputs()
192 …es->ptr_to_element(Coordinates(1, i)))) = *(reinterpret_cast<float *>(decoded_boxes->ptr_to_elemen… in SaveOutputs()
193 …es->ptr_to_element(Coordinates(2, i)))) = *(reinterpret_cast<float *>(decoded_boxes->ptr_to_elemen… in SaveOutputs()
194 …es->ptr_to_element(Coordinates(3, i)))) = *(reinterpret_cast<float *>(decoded_boxes->ptr_to_elemen… in SaveOutputs()
|