Searched refs:kOutputKeypointTensor (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | HeatmapMaxKeypoint.cpp | 249 Shape outputKeypoint = context->getOutputShape(kOutputKeypointTensor); in prepare()
258 NN_RET_CHECK(context->setOutputShape(kOutputKeypointTensor, outputKeypoint)); in prepare()
272 auto outputKeypointData = context->getOutputBuffer<_Float16>(kOutputKeypointTensor); in execute()
273 const auto outputKeypointShape = context->getOutputShape(kOutputKeypointTensor); in execute()
295 context->getOutputBuffer<float>(kOutputKeypointTensor), in execute()
296 context->getOutputShape(kOutputKeypointTensor), 1e-5f, in execute()
307 context->getOutputBuffer<uint16_t>(kOutputKeypointTensor), in execute()
308 context->getOutputShape(kOutputKeypointTensor), 1e-5f, 1e-5f); in execute()
318 context->getOutputBuffer<uint16_t>(kOutputKeypointTensor), in execute()
319 context->getOutputShape(kOutputKeypointTensor), 1e-5f, 1e-5f); in execute()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | HeatmapMaxKeypoint.h | 33 constexpr uint32_t kOutputKeypointTensor = 1; variable
|