Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DGenerateProposals.cpp211 const uint32_t kRoiDim = 4; in prepare() local
213 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / kRoiDim; in prepare()
217 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), kRoiDim); in prepare()
219 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 1), kRoiDim * numClasses); in prepare()
231 outputShape.dimensions = {numRois, numClasses * kRoiDim}; in prepare()
456 const uint32_t kRoiDim = 4; in boxWithNmsLimitFloat32Compute() local
477 const float* roi = roiBase + i * kRoiDim; in boxWithNmsLimitFloat32Compute()
486 [&roiBase](uint32_t ind) { return roiBase + ind * kRoiDim; }, kernel, &result); in boxWithNmsLimitFloat32Compute()
497 roiBase += batchSplitIn->at(b) * numClasses * kRoiDim; in boxWithNmsLimitFloat32Compute()
522 const uint32_t kRoiDim = 4; in boxWithNmsLimitWriteOutput() local
[all …]
DRoiAlign.cpp54 const uint32_t kRoiDim = 4; in roiAlignNhwc() local
70 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignNhwc()
181 constexpr uint32_t kRoiDim = 4; in roiAlignQuantNhwc() local
197 for (const uint16_t* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignQuantNhwc()
DRoiPooling.cpp48 const uint32_t kRoiDim = 4; in roiPoolingNhwc() local
64 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiPoolingNhwc()