Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/
DBoundingBox.cpp73 inline void fillRoiTensor(uint32_t numRois, T maxH, T maxW, RandomOperand* op) { in fillRoiTensor() argument
74 for (uint32_t i = 0; i < numRois; i++) { in fillRoiTensor()
95 uint32_t numRois = op->inputs[1]->dimensions[0].getValue(); in roiFinalizer() local
101 fillRoiTensor<float>(numRois, maxH, maxW, op->inputs[1].get()); in roiFinalizer()
106 fillRoiTensor<uint16_t>(numRois, maxH, maxW, op->inputs[1].get()); in roiFinalizer()
114 std::vector<int32_t> batchIndex(numRois); in roiFinalizer()
115 for (uint32_t i = 0; i < numRois; i++) batchIndex[i] = getUniform<int32_t>(0, batch - 1); in roiFinalizer()
118 for (uint32_t i = 0; i < numRois; i++) op->inputs[2]->value<int32_t>(i) = batchIndex[i]; in roiFinalizer()
182 RandomVariable numRois = RandomVariableType::FREE; in heatmapMaxKeypointConstructor() local
187 op->inputs[0]->dimensions = {numRois, numKeypoints, heatmapSize, heatmapSize}; in heatmapMaxKeypointConstructor()
[all …]
/frameworks/ml/nn/common/operations/
DRoiPooling.cpp70 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiPoolingNhwc() local
74 const T_Roi* roiDataEnd = roiData + numRois * roiInfoLength; in roiPoolingNhwc()
241 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
243 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
267 output.dimensions = {numRois, inDepth, static_cast<uint32_t>(outputHeight), in prepare()
270 output.dimensions = {numRois, static_cast<uint32_t>(outputHeight), in prepare()
DRoiAlign.cpp74 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc() local
78 const T_Roi* roiDataEnd = roiData + numRois * roiInfoLength; in roiAlignNhwc()
201 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignQuantNhwc() local
205 const uint16_t* roiDataEnd = roiData + numRois * roiInfoLength; in roiAlignQuantNhwc()
396 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
403 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
433 output.dimensions = {numRois, inDepth, static_cast<uint32_t>(outputHeight), in prepare()
436 output.dimensions = {numRois, static_cast<uint32_t>(outputHeight), in prepare()
DGenerateProposals.cpp243 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
249 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 0), numRois); in prepare()
251 NN_RET_CHECK_EQ(getSizeOfDimension(batchesShape, 0), numRois); in prepare()
262 outputShape.dimensions = {numRois, numClasses * kRoiDim}; in prepare()
387 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass() argument
395 for (uint32_t b = 0; b < numRois; b++) { in hardNmsMultiClass()
451 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass() argument
459 for (uint32_t b = 0; b < numRois; b++) { in softNmsMultiClass()
505 uint32_t numRois = getSizeOfDimension(scoresShape, 0); in boxWithNmsLimitFloat32Compute() local
510 for (uint32_t i = 0, ind = -1; i < numRois; i++) { in boxWithNmsLimitFloat32Compute()
[all …]