Lines Matching refs:kOutputTensor
31 constexpr uint32_t kOutputTensor = 0; variable
83 const OperandType outputType = context->getOutputType(kOutputTensor); in validate()
109 Shape output = context->getOutputShape(kOutputTensor); in prepare()
111 return context->setOutputShape(kOutputTensor, output); in prepare()
116 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
119 const OperandType outputType = context->getOutputType(kOutputTensor); in execute()
126 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
128 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
134 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
136 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
142 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
144 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
150 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
153 context->getOutputBuffer<float>(kOutputTensor)); in execute()