Lines Matching refs:numBatches
68 uint32_t numBatches = getSizeOfDimension(imageInfoDataShape, 0); in bboxTransformFloat32() local
80 NN_RET_CHECK_LT(batchIndex, numBatches); in bboxTransformFloat32()
245 uint32_t numBatches = getSizeOfDimension(imageInfoShape, 0); in prepare() local
247 NN_RET_CHECK_GT(numBatches, 0); in prepare()
962 uint32_t numBatches = getSizeOfDimension(scoresShape, 0); in generateProposalsNhwcFloat32Compute() local
1005 for (uint32_t b = 0; b < numBatches; b++) { in generateProposalsNhwcFloat32Compute()
1296 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
1301 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 0), numBatches); in prepare()
1305 NN_RET_CHECK_EQ(getSizeOfDimension(imageInfoDataShape, 0), numBatches); in prepare()
1449 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in detectionPostprocessFloat32() local
1467 for (uint32_t b = 0; b < numBatches; b++) { in detectionPostprocessFloat32()
1620 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
1629 NN_RET_CHECK_EQ(getSizeOfDimension(deltasShape, 0), numBatches); in prepare()
1662 outputScoreShape.dimensions = {numBatches, numOutDetections}; in prepare()
1666 outputRoiShape.dimensions = {numBatches, numOutDetections, 4}; in prepare()
1670 outputClassShape.dimensions = {numBatches, numOutDetections}; in prepare()
1674 outputDetectionShape.dimensions = {numBatches}; in prepare()