Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DGenerateProposals.cpp206 const uint32_t kRoiDim = 4; in prepare() local
208 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / kRoiDim; in prepare()
212 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), kRoiDim); in prepare()
214 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 1), kRoiDim * numClasses); in prepare()
226 outputShape.dimensions = {numRois, numClasses * kRoiDim}; in prepare()
452 const uint32_t kRoiDim = 4; in boxWithNmsLimitFloat32Compute() local
472 const float* roi = roiBase + i * kRoiDim; in boxWithNmsLimitFloat32Compute()
480 [&roiBase](uint32_t ind) { return roiBase + ind * kRoiDim; }, kernel, in boxWithNmsLimitFloat32Compute()
492 roiBase += batchSplitIn->at(b) * numClasses * kRoiDim; in boxWithNmsLimitFloat32Compute()
515 const uint32_t kRoiDim = 4; in boxWithNmsLimitWriteOutput() local
[all …]
DRoiAlign.cpp58 const uint32_t kRoiDim = 4; in roiAlignNhwc() local
74 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignNhwc()
186 constexpr uint32_t kRoiDim = 4; in roiAlignNhwc() local
202 for (const uint16_t* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignNhwc()
DRoiPooling.cpp54 const uint32_t kRoiDim = 4; in roiPoolingNhwc() local
70 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiPoolingNhwc()