Home
last modified time | relevance | path

Searched refs:numRois (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/
DBoundingBox.cpp75 inline void fillRoiTensor(uint32_t numRois, T maxH, T maxW, RandomOperand* op) { in fillRoiTensor() argument
78 for (uint32_t i = 0; i < numRois; i++) { in fillRoiTensor()
99 uint32_t numRois = op->inputs[1]->dimensions[0].getValue(); in roiFinalizer() local
105 fillRoiTensor<float>(numRois, maxH, maxW, op->inputs[1].get()); in roiFinalizer()
110 fillRoiTensor<uint16_t>(numRois, maxH, maxW, op->inputs[1].get()); in roiFinalizer()
118 std::vector<int32_t> batchIndex(numRois); in roiFinalizer()
119 for (uint32_t i = 0; i < numRois; i++) batchIndex[i] = getUniform<int32_t>(0, batch - 1); in roiFinalizer()
122 for (uint32_t i = 0; i < numRois; i++) op->inputs[2]->value<int32_t>(i) = batchIndex[i]; in roiFinalizer()
187 RandomVariable numRois = RandomVariableType::FREE; in heatmapMaxKeypointConstructor() local
192 op->inputs[0]->dimensions = {numRois, numKeypoints, heatmapSize, heatmapSize}; in heatmapMaxKeypointConstructor()
[all …]
/packages/modules/NeuralNetworks/common/cpu_operations/
DRoiPooling.cpp58 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiPoolingNhwc() local
62 const T_Roi* roiDataEnd = roiData + numRois * roiInfoLength; in roiPoolingNhwc()
190 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
192 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
216 output.dimensions = {numRois, inDepth, static_cast<uint32_t>(outputHeight), in prepare()
219 output.dimensions = {numRois, static_cast<uint32_t>(outputHeight), in prepare()
DRoiAlign.cpp64 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc() local
68 const T_Roi* roiDataEnd = roiData + numRois * roiInfoLength; in roiAlignNhwc()
191 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignQuantNhwc() local
195 const uint16_t* roiDataEnd = roiData + numRois * roiInfoLength; in roiAlignQuantNhwc()
344 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
351 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
381 output.dimensions = {numRois, inDepth, static_cast<uint32_t>(outputHeight), in prepare()
384 output.dimensions = {numRois, static_cast<uint32_t>(outputHeight), in prepare()
DGenerateProposals.cpp212 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
218 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 0), numRois); in prepare()
220 NN_RET_CHECK_EQ(getSizeOfDimension(batchesShape, 0), numRois); in prepare()
231 outputShape.dimensions = {numRois, numClasses * kRoiDim}; in prepare()
339 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass() argument
347 for (uint32_t b = 0; b < numRois; b++) { in hardNmsMultiClass()
403 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass() argument
411 for (uint32_t b = 0; b < numRois; b++) { in softNmsMultiClass()
457 uint32_t numRois = getSizeOfDimension(scoresShape, 0); in boxWithNmsLimitFloat32Compute() local
463 for (uint32_t i = 0; i < numRois; i++) { in boxWithNmsLimitFloat32Compute()
[all …]