Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DRoiPooling.cpp43 const Shape& roiShape, const int32_t* batchSplitData, in roiPoolingNhwc() argument
58 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiPoolingNhwc()
59 uint32_t roiInfoLength = getSizeOfDimension(roiShape, 1); in roiPoolingNhwc()
132 const Shape& roiShape, const int32_t* batchSplitData, 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
153 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in roiPooling()
154 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in roiPooling()
155 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData, in roiPooling()
163 const uint16_t* roiData, const Shape& roiShape, in roiPooling() argument
[all …]
DRoiAlign.cpp48 const Shape& roiShape, const int32_t* batchSplitData, in roiAlignNhwc() argument
64 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc()
65 uint32_t roiInfoLength = getSizeOfDimension(roiShape, 1); in roiAlignNhwc()
173 const uint16_t* roiData, const Shape& roiShape, in roiAlignQuantNhwc() argument
191 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignQuantNhwc()
192 uint32_t roiInfoLength = getSizeOfDimension(roiShape, 1); in roiAlignQuantNhwc()
306 const Shape& roiShape, const int32_t* batchSplitData, 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()
335 Shape roiShape = context->getInputShape(kRoiTensor); in prepare() local
[all …]
DGenerateProposals.cpp63 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
[all …]