Lines Matching refs:kOutputTensor
45 constexpr uint32_t kOutputTensor = 0; variable
273 Shape output = context->getOutputShape(kOutputTensor); in prepare()
275 return context->setOutputShape(kOutputTensor, output); in prepare()
280 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
289 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
290 context->getOutputShape(kOutputTensor)); in execute()
295 context->getOutputBuffer<float>(kOutputTensor), in execute()
296 context->getOutputShape(kOutputTensor)); in execute()
301 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
302 context->getOutputShape(kOutputTensor)); in execute()
307 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
308 context->getOutputShape(kOutputTensor)); in execute()