Lines Matching refs:roiShape

63 inline bool bboxTransformFloat32(const float* roiData, const Shape& roiShape,  in bboxTransformFloat32()  argument
74 const float* roiDataEnd = roiData + getNumberOfElements(roiShape); in bboxTransformFloat32()
113 inline bool bboxTransformFloat16(const _Float16* roiData, const Shape& roiShape, in bboxTransformFloat16() argument
118 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in bboxTransformFloat16()
125 NN_RET_CHECK(bboxTransformFloat32(roi_float32.data(), roiShape, delta_float32.data(), in bboxTransformFloat16()
133 inline bool bboxTransformQuant(const uint16_t* roiData, const Shape& roiShape, in bboxTransformQuant() argument
138 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in bboxTransformQuant()
139 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in bboxTransformQuant()
147 NN_RET_CHECK(bboxTransformFloat32(roi_float32.data(), roiShape, delta_float32.data(), in bboxTransformQuant()
155 inline bool bboxTransformQuant(const uint16_t* roiData, const Shape& roiShape, in bboxTransformQuant() argument
160 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in bboxTransformQuant()
161 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in bboxTransformQuant()
169 NN_RET_CHECK(bboxTransformFloat32(roi_float32.data(), roiShape, delta_float32.data(), in bboxTransformQuant()
199 Shape roiShape = context->getInputShape(kRoiTensor); in prepare() local
205 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2u); in prepare()
212 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare()
217 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), kRoiDim); in prepare()
223 if (roiShape.type == OperandType::TENSOR_QUANT16_ASYMM) { in prepare()
224 NN_RET_CHECK_EQ(roiShape.scale, 0.125f); in prepare()
225 NN_RET_CHECK_EQ(roiShape.offset, 0); in prepare()
230 outputShape.type = roiShape.type; in prepare()
234 if (roiShape.type == OperandType::TENSOR_QUANT16_ASYMM) { in prepare()
571 const Shape& roiShape, const int32_t* batchesData, in boxWithNmsLimitFloat32() argument
586 scores_float32.data(), scoresShape, roiData, roiShape, batchesData, batchesShape, in boxWithNmsLimitFloat32()
594 const _Float16* roiData, const Shape& roiShape, in boxWithNmsLimitFloat16() argument
606 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in boxWithNmsLimitFloat16()
610 scores_float32.data(), scoresShape, roi_float32.data(), roiShape, batchesData, in boxWithNmsLimitFloat16()
618 const uint16_t* roiData, const Shape& roiShape, in boxWithNmsLimitQuant() argument
629 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in boxWithNmsLimitQuant()
630 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in boxWithNmsLimitQuant()
633 scores_float32.data(), scoresShape, roi_float32.data(), roiShape, batchesData, in boxWithNmsLimitQuant()
641 const uint16_t* roiData, const Shape& roiShape, in boxWithNmsLimitQuant() argument
653 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in boxWithNmsLimitQuant()
654 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in boxWithNmsLimitQuant()
657 scores_float32.data(), scoresShape, roi_float32.data(), roiShape, batchesData, in boxWithNmsLimitQuant()
668 Shape roiShape = context->getInputShape(kRoiTensor); in prepare() local
676 NN_RET_CHECK(getNumberOfDimensions(roiShape) == 2); in prepare()
683 NN_RET_CHECK(getSizeOfDimension(roiShape, 0) == numRois); in prepare()
684 NN_RET_CHECK(getSizeOfDimension(roiShape, 1) == kRoiDim * numClasses); in prepare()
690 NN_RET_CHECK_EQ(roiShape.scale, 0.125f); in prepare()
691 NN_RET_CHECK_EQ(roiShape.offset, 0); in prepare()
700 outputRoiShape.type = roiShape.type; in prepare()