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.cpp67 inline void fillRoiTensor(uint32_t numRois, T maxH, T maxW, RandomOperand* op) { in fillRoiTensor() argument
68 for (uint32_t i = 0; i < numRois; i++) { in fillRoiTensor()
89 uint32_t numRois = op->inputs[1]->dimensions[0].getValue(); in roiFinalizer() local
95 fillRoiTensor<float>(numRois, maxH, maxW, op->inputs[1].get()); in roiFinalizer()
100 fillRoiTensor<uint16_t>(numRois, maxH, maxW, op->inputs[1].get()); in roiFinalizer()
108 std::vector<int32_t> batchIndex(numRois); in roiFinalizer()
109 for (uint32_t i = 0; i < numRois; i++) batchIndex[i] = getUniform<int32_t>(0, batch - 1); in roiFinalizer()
112 for (uint32_t i = 0; i < numRois; i++) op->inputs[2]->value<int32_t>(i) = batchIndex[i]; in roiFinalizer()
166 RandomVariable numRois = RandomVariableType::FREE; in heatmapMaxKeypointConstructor() local
171 op->inputs[0]->dimensions = {numRois, numKeypoints, heatmapSize, heatmapSize}; in heatmapMaxKeypointConstructor()
[all …]
/frameworks/ml/nn/common/operations/
DRoiPooling.cpp64 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiPoolingNhwc() local
68 const T_Roi* roiDataEnd = roiData + numRois * roiInfoLength; in roiPoolingNhwc()
214 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
216 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
240 output.dimensions = {numRois, inDepth, static_cast<uint32_t>(outputHeight), in prepare()
243 output.dimensions = {numRois, static_cast<uint32_t>(outputHeight), in prepare()
DRoiAlign.cpp68 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc() local
72 const T_Roi* roiDataEnd = roiData + numRois * roiInfoLength; in roiAlignNhwc()
196 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc() local
200 const uint16_t* roiDataEnd = roiData + numRois * roiInfoLength; in roiAlignNhwc()
379 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
386 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
416 output.dimensions = {numRois, inDepth, static_cast<uint32_t>(outputHeight), in prepare()
419 output.dimensions = {numRois, static_cast<uint32_t>(outputHeight), in prepare()
DGenerateProposals.cpp207 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
213 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 0), numRois); in prepare()
215 NN_RET_CHECK_EQ(getSizeOfDimension(batchesShape, 0), numRois); in prepare()
226 outputShape.dimensions = {numRois, numClasses * kRoiDim}; in prepare()
335 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass() argument
343 for (uint32_t b = 0; b < numRois; b++) { in hardNmsMultiClass()
399 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass() argument
407 for (uint32_t b = 0; b < numRois; b++) { in softNmsMultiClass()
453 uint32_t numRois = getSizeOfDimension(scoresShape, 0); in boxWithNmsLimitFloat32Compute() local
458 for (uint32_t i = 0, ind = -1; i < numRois; i++) { in boxWithNmsLimitFloat32Compute()
[all …]