Searched refs:numKeypoints (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | HeatmapMaxKeypoint.cpp | 90 uint32_t numKeypoints = getSizeOfDimension(heatmapShape, 3); in heatmapMaxKeypointFloat32Nhwc() local
100 for (uint32_t j = 0; j < numKeypoints; j++) { in heatmapMaxKeypointFloat32Nhwc()
105 float val = heatmapBase[k * numKeypoints + j]; in heatmapMaxKeypointFloat32Nhwc()
128 uint32_t heatmapIndex = static_cast<uint32_t>(h) * heatmapSize * numKeypoints + in heatmapMaxKeypointFloat32Nhwc()
129 static_cast<uint32_t>(w) * numKeypoints + j; in heatmapMaxKeypointFloat32Nhwc()
153 heatmapBase += heatmapSize * heatmapSize * numKeypoints; in heatmapMaxKeypointFloat32Nhwc()
231 uint32_t numKeypoints = getSizeOfDimension(heatmapShape, layout ? 1 : 3); in prepare() local
246 outputScore.dimensions = {numBoxes, numKeypoints}; in prepare()
251 outputKeypoint.dimensions = {numBoxes, numKeypoints, 2}; in prepare()
|
| /packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/ |
| D | BoundingBox.cpp | 188 RandomVariable numKeypoints = RandomVariableType::FREE; in heatmapMaxKeypointConstructor() local
192 op->inputs[0]->dimensions = {numRois, numKeypoints, heatmapSize, heatmapSize}; in heatmapMaxKeypointConstructor()
194 op->inputs[0]->dimensions = {numRois, heatmapSize, heatmapSize, numKeypoints}; in heatmapMaxKeypointConstructor()
197 op->outputs[0]->dimensions = {numRois, numKeypoints}; in heatmapMaxKeypointConstructor()
198 op->outputs[1]->dimensions = {numRois, numKeypoints, 2}; in heatmapMaxKeypointConstructor()
208 {numRois, numKeypoints, heatmapSize * heatmapSize}); in heatmapMaxKeypointConstructor()
|