Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/
DBoundingBox.cpp118 std::vector<int32_t> batchIndex(numRois); in roiFinalizer() local
119 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()
/packages/modules/NeuralNetworks/tools/test_generator/test_harness/
DTestHarness.cpp179 const int batchIndex = kBatchSize * b; in expectMultinomialDistributionWithinTolerance() local
181 probabilitySum += expf(inputData[batchIndex + i]); in expectMultinomialDistributionWithinTolerance()
186 float probabilityExpected = expf(inputData[batchIndex + i]) / probabilitySum; in expectMultinomialDistributionWithinTolerance()
/packages/modules/NeuralNetworks/common/cpu_operations/
DGenerateProposals.cpp79 uint32_t batchIndex = batchesData[roiIndex]; in bboxTransformFloat32() local
83 NN_RET_CHECK_GE(batchIndex, 0u); in bboxTransformFloat32()
84 NN_RET_CHECK_LT(batchIndex, numBatches); in bboxTransformFloat32()
88 const float* imageInfoBase = imageInfoData + batchIndex * imageLength; in bboxTransformFloat32()