Home
last modified time | relevance | path

Searched refs:BoxCornerEncoding (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Ddetection_postprocess.cc70 struct BoxCornerEncoding { struct
84 static_assert(sizeof(BoxCornerEncoding) == sizeof(float) * kNumCoordBox,
345 auto& box = ReInterpretTensor<BoxCornerEncoding*>(decoded_boxes)[idx]; in DecodeCenterSizeBoxes()
390 auto& box = ReInterpretTensor<const BoxCornerEncoding*>(decoded_boxes)[i]; in ValidateBoxes()
403 auto& box_i = ReInterpretTensor<const BoxCornerEncoding*>(decoded_boxes)[i]; in ComputeIntersectionOverUnion()
404 auto& box_j = ReInterpretTensor<const BoxCornerEncoding*>(decoded_boxes)[j]; in ComputeIntersectionOverUnion()
714 ReInterpretTensor<BoxCornerEncoding*>(detection_boxes)[output_box_index] = in NonMaxSuppressionMultiClassRegularHelper()
715 ReInterpretTensor<const BoxCornerEncoding*>( in NonMaxSuppressionMultiClassRegularHelper()
723 ReInterpretTensor<BoxCornerEncoding*>( in NonMaxSuppressionMultiClassRegularHelper()
815 ReInterpretTensor<BoxCornerEncoding*>(detection_boxes)[box_offset] = in NonMaxSuppressionMultiClassFastHelper()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dnon_max_suppression.h27 struct BoxCornerEncoding { struct
36 auto& box_i = reinterpret_cast<const BoxCornerEncoding*>(boxes)[i]; in ComputeIntersectionOverUnion() argument
37 auto& box_j = reinterpret_cast<const BoxCornerEncoding*>(boxes)[j]; in ComputeIntersectionOverUnion()