Searched refs:batchIndex (Results 1 – 5 of 5) 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()
634 ssize_t batchIndex = findBatch(mMsg.body.motion.deviceId, mMsg.body.motion.source); in consume() local635 if (batchIndex >= 0) { in consume()636 Batch& batch = mBatches.editItemAt(batchIndex); in consume()653 mBatches.removeAt(batchIndex); in consume()660 mBatches.removeAt(batchIndex); in consume()
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()
75 uint32_t batchIndex = batchesData[roiIndex]; in bboxTransformFloat32() local79 NN_RET_CHECK_GE(batchIndex, 0); in bboxTransformFloat32()80 NN_RET_CHECK_LT(batchIndex, numBatches); in bboxTransformFloat32()84 const float* imageInfoBase = imageInfoData + batchIndex * imageLength; in bboxTransformFloat32()
1099 for (int batchIndex = mAlarmBatches.size() - 1; batchIndex >= 0; batchIndex--) { in reorderAlarmsBasedOnStandbyBuckets()1100 final Batch batch = mAlarmBatches.get(batchIndex); in reorderAlarmsBasedOnStandbyBuckets()1114 mAlarmBatches.remove(batchIndex); in reorderAlarmsBasedOnStandbyBuckets()