Searched defs:heatmapShape (Results 1 – 1 of 1) sorted by relevance
89 inline bool heatmapMaxKeypointFloat32Nhwc(const float* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointFloat32Nhwc()168 inline bool heatmapMaxKeypointFloat32(const float* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointFloat32()185 inline bool heatmapMaxKeypointQuant(const uint8_t* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointQuant()207 inline bool heatmapMaxKeypointQuant(const int8_t* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointQuant()263 Shape heatmapShape = context->getInputShape(kHeatmapTensor); in prepare() local306 const auto heatmapShape = context->getInputShape(kHeatmapTensor); in execute() local