Lines Matching refs:kOutputTensor
47 constexpr uint32_t kOutputTensor = 0; variable
536 Shape output = context->getOutputShape(kOutputTensor); in prepare()
543 return context->setOutputShape(kOutputTensor, output); in prepare()
548 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
559 context->getOutputBuffer<float>(kOutputTensor), in execute()
560 context->getOutputShape(kOutputTensor)); in execute()
568 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
569 context->getOutputShape(kOutputTensor)); in execute()
583 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
584 context->getOutputShape(kOutputTensor)); in execute()
592 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
593 context->getOutputShape(kOutputTensor)); in execute()
610 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
611 context->getOutputShape(kOutputTensor)); in execute()
620 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
621 context->getOutputShape(kOutputTensor)); in execute()