Lines Matching refs:kOutputTensor

43 constexpr uint32_t kOutputTensor = 0;  variable
123     Shape output = context->getOutputShape(kOutputTensor);  in prepare()
125     return context->setOutputShape(kOutputTensor, output);  in prepare()
139                     context->getOutputBuffer<_Float16>(kOutputTensor),  in execute()
140                     context->getOutputShape(kOutputTensor));  in execute()
150                     context->getOutputBuffer<float>(kOutputTensor),  in execute()
151                     context->getOutputShape(kOutputTensor));  in execute()
157                               context->getOutputBuffer<uint8_t>(kOutputTensor),  in execute()
158                               context->getOutputShape(kOutputTensor));  in execute()
165                               context->getOutputBuffer<int8_t>(kOutputTensor),  in execute()
166                               context->getOutputShape(kOutputTensor));  in execute()