Searched refs:kOutputKeypointTensor (Results 1 – 1 of 1) sorted by relevance
45 constexpr uint32_t kOutputKeypointTensor = 1; variable288 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()[all …]