Lines Matching refs:kOutputTensor
35 constexpr uint32_t kOutputTensor = 0; variable
71 const OperandType outputType = context->getOutputType(kOutputTensor); in validate()
88 Shape output = context->getOutputShape(kOutputTensor); in prepare()
90 return context->setOutputShape(kOutputTensor, output); in prepare()
95 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
98 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
102 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
103 context->getOutputShape(kOutputTensor)); in execute()
106 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
107 context->getOutputShape(kOutputTensor)); in execute()
112 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
113 context->getOutputShape(kOutputTensor)); in execute()
116 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
117 context->getOutputShape(kOutputTensor)); in execute()
121 << inputType << " output type: " << context->getOutputType(kOutputTensor) in execute()