Lines Matching refs:kOutputKeypointTensor
45 constexpr uint32_t kOutputKeypointTensor = 1; variable
288 Shape outputKeypoint = context->getOutputShape(kOutputKeypointTensor); in prepare()
297 NN_RET_CHECK(context->setOutputShape(kOutputKeypointTensor, outputKeypoint)); in prepare()
311 auto outputKeypointData = context->getOutputBuffer<_Float16>(kOutputKeypointTensor); in execute()
312 const auto outputKeypointShape = context->getOutputShape(kOutputKeypointTensor); in execute()
334 context->getOutputBuffer<float>(kOutputKeypointTensor), in execute()
335 context->getOutputShape(kOutputKeypointTensor), 1e-5f, in execute()
346 context->getOutputBuffer<uint16_t>(kOutputKeypointTensor), in execute()
347 context->getOutputShape(kOutputKeypointTensor), 1e-5f, 1e-5f); in execute()
357 context->getOutputBuffer<uint16_t>(kOutputKeypointTensor), in execute()
358 context->getOutputShape(kOutputKeypointTensor), 1e-5f, 1e-5f); in execute()