Searched refs:heatmap (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | HeatmapMaxKeypoint.cpp | 80 inline bool heatmapMaxKeypointFloat32Nhwc(const float* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointFloat32Nhwc() argument
93 const float* heatmapBase = heatmap; in heatmapMaxKeypointFloat32Nhwc()
159 inline bool heatmapMaxKeypointFloat32(const float* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointFloat32() argument
167 NN_RET_CHECK(convertNchwToNhwc(heatmap, heatmapShape, &heatmap_nhwc, &heatmapShape_nhwc)); in heatmapMaxKeypointFloat32()
169 const float* heatmap_tmp = layout ? heatmap_nhwc.data() : heatmap; in heatmapMaxKeypointFloat32()
176 inline bool heatmapMaxKeypointQuant(const uint8_t* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointQuant() argument
182 convertQuantToFloat32(heatmap, heatmapShape.scale, heatmapShape.offset, &heatmap_float32); in heatmapMaxKeypointQuant()
198 inline bool heatmapMaxKeypointQuant(const int8_t* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointQuant() argument
204 convertQuantToFloat32(heatmap, heatmapShape.scale, heatmapShape.offset, &heatmap_float32); in heatmapMaxKeypointQuant()
266 const auto heatmap = context->getInputBuffer<_Float16>(kHeatmapTensor); in execute() local
[all …]
|