Searched refs:BoxCornerEncoding (Results 1 – 1 of 1) sorted by relevance
61 struct BoxCornerEncoding { struct75 static_assert(sizeof(BoxCornerEncoding) == sizeof(float) * kNumCoordBox,309 auto& box = ReInterpretTensor<BoxCornerEncoding*>(decoded_boxes)[idx]; in DecodeCenterSizeBoxes()341 auto& box = ReInterpretTensor<const BoxCornerEncoding*>(decoded_boxes)[i]; in ValidateBoxes()351 auto& box_i = ReInterpretTensor<const BoxCornerEncoding*>(decoded_boxes)[i]; in ComputeIntersectionOverUnion()352 auto& box_j = ReInterpretTensor<const BoxCornerEncoding*>(decoded_boxes)[j]; in ComputeIntersectionOverUnion()547 ReInterpretTensor<BoxCornerEncoding*>(detection_boxes)[output_box_index] = in NonMaxSuppressionMultiClassRegularHelper()548 ReInterpretTensor<const BoxCornerEncoding*>( in NonMaxSuppressionMultiClassRegularHelper()555 ReInterpretTensor<BoxCornerEncoding*>( in NonMaxSuppressionMultiClassRegularHelper()631 ReInterpretTensor<BoxCornerEncoding*>(detection_boxes)[box_offset] = in NonMaxSuppressionMultiClassFastHelper()[all …]