Home
last modified time | relevance | path

Searched refs:kBoxesTensor (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DHeatmapMaxKeypoint.cpp225 Shape boxesShape = context->getInputShape(kBoxesTensor); in prepare()
268 const auto boxes = context->getInputBuffer<_Float16>(kBoxesTensor); in execute()
269 const auto boxesShape = context->getInputShape(kBoxesTensor); in execute()
291 context->getInputBuffer<float>(kBoxesTensor), in execute()
292 context->getInputShape(kBoxesTensor), layout, in execute()
303 context->getInputBuffer<uint16_t>(kBoxesTensor), in execute()
304 context->getInputShape(kBoxesTensor), layout, in execute()
314 context->getInputBuffer<uint16_t>(kBoxesTensor), in execute()
315 context->getInputShape(kBoxesTensor), layout, in execute()
/packages/modules/NeuralNetworks/common/types/operations/include/
DHeatmapMaxKeypoint.h28 constexpr uint32_t kBoxesTensor = 1; variable