Lines Matching refs:kOutputTensor
46 constexpr uint32_t kOutputTensor = 0; variable
254 const float outputScale = context->getOutputShape(kOutputTensor).scale; in validate()
299 Shape output = context->getOutputShape(kOutputTensor); in prepare()
301 return context->setOutputShape(kOutputTensor, output); in prepare()
306 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
316 context->getOutputBuffer<float>(kOutputTensor), in execute()
317 context->getOutputShape(kOutputTensor)); in execute()
326 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
327 context->getOutputShape(kOutputTensor)); in execute()
336 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
337 context->getOutputShape(kOutputTensor)); in execute()
346 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
347 context->getOutputShape(kOutputTensor)); in execute()