Searched refs:heatmapSize (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | HeatmapMaxKeypoint.cpp | 89 uint32_t heatmapSize = getSizeOfDimension(heatmapShape, 1); in heatmapMaxKeypointFloat32Nhwc() local
104 for (uint32_t k = 0; k < heatmapSize * heatmapSize; k++) { in heatmapMaxKeypointFloat32Nhwc()
112 uint32_t maxIndexWidth = maxIndex % heatmapSize; in heatmapMaxKeypointFloat32Nhwc()
113 uint32_t maxIndexHeight = maxIndex / heatmapSize; in heatmapMaxKeypointFloat32Nhwc()
125 h = h < 0 ? 1 : (static_cast<uint32_t>(h) >= heatmapSize ? heatmapSize - 2 : h); in heatmapMaxKeypointFloat32Nhwc()
126 w = w < 0 ? 1 : (static_cast<uint32_t>(w) >= heatmapSize ? heatmapSize - 2 : w); in heatmapMaxKeypointFloat32Nhwc()
128 uint32_t heatmapIndex = static_cast<uint32_t>(h) * heatmapSize * numKeypoints + in heatmapMaxKeypointFloat32Nhwc()
144 static_cast<float>(heatmapSize); in heatmapMaxKeypointFloat32Nhwc()
146 static_cast<float>(heatmapSize); in heatmapMaxKeypointFloat32Nhwc()
153 heatmapBase += heatmapSize * heatmapSize * numKeypoints; in heatmapMaxKeypointFloat32Nhwc()
[all …]
|
| /packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/ |
| D | BoundingBox.cpp | 186 RandomVariable heatmapSize = RandomVariableType::FREE; in heatmapMaxKeypointConstructor() local
189 heatmapSize.setRange(2, kInvalidValue); in heatmapMaxKeypointConstructor()
192 op->inputs[0]->dimensions = {numRois, numKeypoints, heatmapSize, heatmapSize}; in heatmapMaxKeypointConstructor()
194 op->inputs[0]->dimensions = {numRois, heatmapSize, heatmapSize, numKeypoints}; in heatmapMaxKeypointConstructor()
208 {numRois, numKeypoints, heatmapSize * heatmapSize}); in heatmapMaxKeypointConstructor()
213 uint32_t heatmapSize = op->inputs[0]->dimensions[2].getValue(); in heatmapMaxKeypointFinalizer() local
217 float maxSize = heatmapSize; in heatmapMaxKeypointFinalizer()
221 uint16_t maxSize = static_cast<uint16_t>(heatmapSize * 8); in heatmapMaxKeypointFinalizer()
|