Lines Matching refs:kRoiDim
211 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
558 memcpy(roiOutPtr, roiBase + index * kRoiDim, kRoiDim * sizeof(T_Roi)); in boxWithNmsLimitWriteOutput()
559 roiOutPtr += kRoiDim; in boxWithNmsLimitWriteOutput()
564 roiBase += batchSplitIn[b] * numClasses * kRoiDim; in boxWithNmsLimitWriteOutput()
680 const uint32_t kRoiDim = 4; in prepare() local
684 NN_RET_CHECK(getSizeOfDimension(roiShape, 1) == kRoiDim * numClasses); in prepare()
829 const uint32_t kRoiDim = 4; in filterBoxes() local
832 const float* roiInfo = roiBase + (*select)[j] * kRoiDim; in filterBoxes()
855 const uint32_t kRoiDim = 4; in generateProposalsNhwcFloat32Compute() local
863 uint32_t roiBufferSize = batchSize * kRoiDim; in generateProposalsNhwcFloat32Compute()
877 for (uint32_t a = 0; a < numAnchors; a++, roiBase += kRoiDim, anchorsBase += kRoiDim) { in generateProposalsNhwcFloat32Compute()
891 tempRoiShape.dimensions = {batchSize, kRoiDim}; in generateProposalsNhwcFloat32Compute()
893 tempBBoxDeltasShape.dimensions = {batchSize, kRoiDim}; in generateProposalsNhwcFloat32Compute()
928 return roiTransformedBuffer.data() + ind * kRoiDim; in generateProposalsNhwcFloat32Compute()
936 roiOutData->insert(roiOutData->end(), roiTransformedBuffer.begin() + i * kRoiDim, in generateProposalsNhwcFloat32Compute()
937 roiTransformedBuffer.begin() + (i + 1) * kRoiDim); in generateProposalsNhwcFloat32Compute()
1128 const uint32_t kRoiDim = 4; in prepare() local
1137 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, useNchw ? 1 : 3), numAnchors * kRoiDim); in prepare()
1141 NN_RET_CHECK_EQ(getSizeOfDimension(anchorsShape, 1), kRoiDim); in prepare()
1262 const uint32_t kRoiDim = 4; in detectionPostprocessFloat32() local
1279 std::vector<float> roiBuffer(numAnchors * kRoiDim); in detectionPostprocessFloat32()
1288 roiBuffer[a * kRoiDim] = yCtr - hHalf; in detectionPostprocessFloat32()
1289 roiBuffer[a * kRoiDim + 1] = xCtr - wHalf; in detectionPostprocessFloat32()
1290 roiBuffer[a * kRoiDim + 2] = yCtr + hHalf; in detectionPostprocessFloat32()
1291 roiBuffer[a * kRoiDim + 3] = xCtr + wHalf; in detectionPostprocessFloat32()
1292 anchorBase += kRoiDim; in detectionPostprocessFloat32()
1302 return roiBuffer.data() + (ind / numClasses) * kRoiDim; in detectionPostprocessFloat32()
1308 memcpy(roiOutBase + i * kRoiDim, &roiBuffer[(ind / numClasses) * kRoiDim], in detectionPostprocessFloat32()
1309 kRoiDim * sizeof(float)); in detectionPostprocessFloat32()
1328 [&roiBuffer](uint32_t ind) { return roiBuffer.data() + ind * kRoiDim; }, in detectionPostprocessFloat32()
1344 memcpy(roiOutPtr, &roiBuffer[i * kRoiDim], kRoiDim * sizeof(float)); in detectionPostprocessFloat32()
1345 roiOutPtr += kRoiDim; in detectionPostprocessFloat32()
1353 roiOutBase += numOutDetection * kRoiDim; in detectionPostprocessFloat32()
1402 const uint32_t kRoiDim = 4; in prepare() local
1415 NN_RET_CHECK_EQ(getSizeOfDimension(anchorsShape, 1), kRoiDim); in prepare()