Searched refs:kHeatmapTensor (Results 1 – 3 of 3) sorted by relevance
224 Shape heatmapShape = context->getInputShape(kHeatmapTensor); in prepare()264 switch (context->getInputType(kHeatmapTensor)) { in execute()266 const auto heatmap = context->getInputBuffer<_Float16>(kHeatmapTensor); in execute()267 const auto heatmapShape = context->getInputShape(kHeatmapTensor); in execute()289 return heatmapMaxKeypointFloat32(context->getInputBuffer<float>(kHeatmapTensor), in execute()290 context->getInputShape(kHeatmapTensor), in execute()301 context->getInputBuffer<uint8_t>(kHeatmapTensor), in execute()302 context->getInputShape(kHeatmapTensor), in execute()312 context->getInputBuffer<int8_t>(kHeatmapTensor), in execute()313 context->getInputShape(kHeatmapTensor), in execute()
27 constexpr uint32_t kHeatmapTensor = 0; variable
31 auto inputType = context->getInputType(kHeatmapTensor); in validate()