Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DGenerateProposals.cpp242 const uint32_t kRoiDim = 4; in prepare() local
244 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / kRoiDim; in prepare()
248 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), kRoiDim); in prepare()
250 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 1), kRoiDim * numClasses); in prepare()
262 outputShape.dimensions = {numRois, numClasses * kRoiDim}; in prepare()
504 const uint32_t kRoiDim = 4; in boxWithNmsLimitFloat32Compute() local
524 const float* roi = roiBase + i * kRoiDim; in boxWithNmsLimitFloat32Compute()
533 [&roiBase](uint32_t ind) { return roiBase + ind * kRoiDim; }, kernel, &result); in boxWithNmsLimitFloat32Compute()
544 roiBase += batchSplitIn->at(b) * numClasses * kRoiDim; in boxWithNmsLimitFloat32Compute()
569 const uint32_t kRoiDim = 4; in boxWithNmsLimitWriteOutput() local
[all …]
DRoiAlign.cpp64 const uint32_t kRoiDim = 4; in roiAlignNhwc() local
80 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignNhwc()
191 constexpr uint32_t kRoiDim = 4; in roiAlignQuantNhwc() local
207 for (const uint16_t* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignQuantNhwc()
DRoiPooling.cpp60 const uint32_t kRoiDim = 4; in roiPoolingNhwc() local
76 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiPoolingNhwc()