Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DRoiAlign.cpp81 uint32_t roiIndex = 0; in roiAlignNhwc() local
82 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignNhwc()
83 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignNhwc()
208 uint32_t roiIndex = 0; in roiAlignQuantNhwc() local
209 for (const uint16_t* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiAlignQuantNhwc()
210 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignQuantNhwc()
H A DRoiPooling.cpp76 uint32_t roiIndex = 0; in roiPoolingNhwc() local
77 for (const T_Roi* roiInfo = roiData; roiInfo < roiDataEnd; roiInfo += kRoiDim, roiIndex++) { in roiPoolingNhwc()
78 uint32_t batchId = batchSplitData[roiIndex]; in roiPoolingNhwc()
H A DGenerateProposals.cpp74 uint32_t roiIndex = 0; in bboxTransformFloat32() local
75 for (const float* roiBase = roiData; roiBase < roiDataEnd; roiBase += roiLength, roiIndex++) { in bboxTransformFloat32()
76 uint32_t batchIndex = batchesData[roiIndex]; in bboxTransformFloat32()