Lines Matching refs:kOutputTensor

32 constexpr uint32_t kOutputTensor = 0;  variable
82     const OperandType outputType = context->getOutputType(kOutputTensor);  in validate()
101     Shape output = context->getOutputShape(kOutputTensor);  in prepare()
103     return context->setOutputShape(kOutputTensor, output);  in prepare()
108     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in execute()
111     const OperandType outputType = context->getOutputType(kOutputTensor);  in execute()
118                            context->getOutputBuffer<_Float16>(kOutputTensor));  in execute()
120             return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(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()
137                                      context->getOutputBuffer<float>(kOutputTensor));  in execute()