Lines Matching refs:kOutputTensor
32 constexpr uint32_t kOutputTensor = 0; variable
84 const OperandType outputType = context->getOutputType(kOutputTensor); in validate()
110 Shape output = context->getOutputShape(kOutputTensor); in prepare()
112 return context->setOutputShape(kOutputTensor, output); in prepare()
117 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
120 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
127 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
129 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
135 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
137 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
143 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
145 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
151 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
154 context->getOutputBuffer<float>(kOutputTensor)); in execute()