Home
last modified time | relevance | path

Searched refs:roiData (Results 1 – 3 of 3) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DRoiPooling.cpp42 inline bool roiPoolingNhwc(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiPoolingNhwc() argument
62 const T_Roi* roiDataEnd = roiData + numRois * roiInfoLength; in roiPoolingNhwc()
64 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiPoolingNhwc()
131 inline bool roiPooling(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiPooling() argument
139 NN_RET_CHECK(roiPoolingNhwc(input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, in roiPooling()
148 const uint16_t* roiData, const Shape& roiShape, in roiPooling() argument
154 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in roiPooling()
163 const uint16_t* roiData, const Shape& roiShape, in roiPooling() argument
169 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in roiPooling()
DRoiAlign.cpp47 inline bool roiAlignNhwc(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiAlignNhwc() argument
68 const T_Roi* roiDataEnd = roiData + numRois * roiInfoLength; in roiAlignNhwc()
70 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignNhwc()
173 const uint16_t* roiData, const Shape& roiShape, in roiAlignQuantNhwc() argument
195 const uint16_t* roiDataEnd = roiData + numRois * roiInfoLength; in roiAlignQuantNhwc()
197 for (const uint16_t* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignQuantNhwc()
305 inline bool roiAlign(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiAlign() argument
317 input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, batchSplitData, in roiAlign()
321 NN_RET_CHECK(roiAlignNhwc(input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, in roiAlign()
DGenerateProposals.cpp63 inline bool bboxTransformFloat32(const float* roiData, const Shape& roiShape, in bboxTransformFloat32() argument
74 const float* roiDataEnd = roiData + getNumberOfElements(roiShape); in bboxTransformFloat32()
78 for (const float* roiBase = roiData; roiBase < roiDataEnd; roiBase += roiLength, roiIndex++) { in bboxTransformFloat32()
113 inline bool bboxTransformFloat16(const _Float16* roiData, const Shape& roiShape, in bboxTransformFloat16() argument
119 convertFloat16ToFloat32(roiData, &roi_float32); in bboxTransformFloat16()
133 inline bool bboxTransformQuant(const uint16_t* roiData, const Shape& roiShape, in bboxTransformQuant() argument
139 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in bboxTransformQuant()
155 inline bool bboxTransformQuant(const uint16_t* roiData, const Shape& roiShape, in bboxTransformQuant() argument
161 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in bboxTransformQuant()
438 const float* roiData, const Shape& /*roiShape*/, in boxWithNmsLimitFloat32Compute() argument
[all …]