Searched refs:batchIndex (Results 1 – 4 of 4) sorted by relevance
118 std::vector<int32_t> batchIndex(numRois); in roiFinalizer() local119 for (uint32_t i = 0; i < numRois; i++) batchIndex[i] = getUniform<int32_t>(0, batch - 1); in roiFinalizer()121 std::sort(batchIndex.begin(), batchIndex.end()); in roiFinalizer()122 for (uint32_t i = 0; i < numRois; i++) op->inputs[2]->value<int32_t>(i) = batchIndex[i]; in roiFinalizer()
179 const int batchIndex = kBatchSize * b; in expectMultinomialDistributionWithinTolerance() local181 probabilitySum += expf(inputData[batchIndex + i]); in expectMultinomialDistributionWithinTolerance()186 float probabilityExpected = expf(inputData[batchIndex + i]) / probabilitySum; in expectMultinomialDistributionWithinTolerance()
765 ssize_t batchIndex = findBatch(mMsg.body.motion.deviceId, mMsg.body.motion.source); in consume() local766 if (batchIndex >= 0) { in consume()767 Batch& batch = mBatches[batchIndex]; in consume()784 mBatches.erase(mBatches.begin() + batchIndex); in consume()791 mBatches.erase(mBatches.begin() + batchIndex); in consume()
76 uint32_t batchIndex = batchesData[roiIndex]; in bboxTransformFloat32() local80 NN_RET_CHECK_GE(batchIndex, 0); in bboxTransformFloat32()81 NN_RET_CHECK_LT(batchIndex, numBatches); in bboxTransformFloat32()85 const float* imageInfoBase = imageInfoData + batchIndex * imageLength; in bboxTransformFloat32()