Searched refs:numKeypoints (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | HeatmapMaxKeypoint.cpp | 99 uint32_t numKeypoints = getSizeOfDimension(heatmapShape, 3); in heatmapMaxKeypointFloat32Nhwc() local
109 for (uint32_t j = 0; j < numKeypoints; j++) { in heatmapMaxKeypointFloat32Nhwc()
114 float val = heatmapBase[k * numKeypoints + j]; in heatmapMaxKeypointFloat32Nhwc()
137 uint32_t heatmapIndex = static_cast<uint32_t>(h) * heatmapSize * numKeypoints + in heatmapMaxKeypointFloat32Nhwc()
138 static_cast<uint32_t>(w) * numKeypoints + j; in heatmapMaxKeypointFloat32Nhwc()
162 heatmapBase += heatmapSize * heatmapSize * numKeypoints; in heatmapMaxKeypointFloat32Nhwc()
270 uint32_t numKeypoints = getSizeOfDimension(heatmapShape, layout ? 1 : 3); in prepare() local
285 outputScore.dimensions = {numBoxes, numKeypoints}; in prepare()
290 outputKeypoint.dimensions = {numBoxes, numKeypoints, 2}; in prepare()
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/ |
| H A 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()
|