Lines Matching refs:kOutputTensor
44 constexpr uint32_t kOutputTensor = 0; variable
474 const float outputScale = context->getInputShape(kOutputTensor).scale; in validate()
568 Shape output = context->getOutputShape(kOutputTensor); in prepare()
575 return context->setOutputShape(kOutputTensor, output); in prepare()
580 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
595 context->getOutputBuffer<float>(kOutputTensor), in execute()
596 context->getOutputShape(kOutputTensor)); in execute()
608 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
609 context->getOutputShape(kOutputTensor)); in execute()
626 param.useNchw, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
627 context->getOutputShape(kOutputTensor)); in execute()
639 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
640 context->getOutputShape(kOutputTensor)); in execute()
660 param.useNchw, context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
661 context->getOutputShape(kOutputTensor)); in execute()
674 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
675 context->getOutputShape(kOutputTensor)); in execute()