Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DRoiAlign.cpp79 uint32_t roiIndex = 0; in roiAlignNhwc() local
80 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignNhwc()
81 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignNhwc()
206 uint32_t roiIndex = 0; in roiAlignQuantNhwc() local
207 for (const uint16_t* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignQuantNhwc()
208 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignQuantNhwc()
DRoiPooling.cpp75 uint32_t roiIndex = 0; in roiPoolingNhwc() local
76 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiPoolingNhwc()
77 uint32_t batchId = batchSplitData[roiIndex]; in roiPoolingNhwc()
DGenerateProposals.cpp73 uint32_t roiIndex = 0; in bboxTransformFloat32() local
74 for (const float* roiBase = roiData; roiBase < roiDataEnd; roiBase += roiLength, roiIndex++) { in bboxTransformFloat32()
75 uint32_t batchIndex = batchesData[roiIndex]; in bboxTransformFloat32()