Lines Matching refs:kOutputTensor
36 constexpr uint32_t kOutputTensor = 0; variable
74 const OperandType outputType = context->getOutputType(kOutputTensor); in validate()
91 Shape output = context->getOutputShape(kOutputTensor); in prepare()
93 return context->setOutputShape(kOutputTensor, output); in prepare()
98 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
101 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
105 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
106 context->getOutputShape(kOutputTensor)); in execute()
109 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
110 context->getOutputShape(kOutputTensor)); in execute()
115 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
116 context->getOutputShape(kOutputTensor)); in execute()
119 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
120 context->getOutputShape(kOutputTensor)); in execute()
125 << " output type: " << toString(context->getOutputType(kOutputTensor)) in execute()