Lines Matching refs:kOutputTensor
48 constexpr uint32_t kOutputTensor = 0; variable
279 Shape output = context->getOutputShape(kOutputTensor); in prepare()
281 return context->setOutputShape(kOutputTensor, output); in prepare()
286 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
295 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
296 context->getOutputShape(kOutputTensor)); in execute()
301 context->getOutputBuffer<float>(kOutputTensor), in execute()
302 context->getOutputShape(kOutputTensor)); in execute()
307 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
308 context->getOutputShape(kOutputTensor)); in execute()
313 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
314 context->getOutputShape(kOutputTensor)); in execute()