Lines Matching refs:kOutputTensor

43 constexpr uint32_t kOutputTensor = 0;  variable
246     Shape output = context->getOutputShape(kOutputTensor);  in prepare()
259     return context->setOutputShape(kOutputTensor, output);  in prepare()
271                                  context->getOutputBuffer<float>(kOutputTensor),  in execute()
272                                  context->getOutputShape(kOutputTensor));  in execute()
276                                  context->getOutputBuffer<_Float16>(kOutputTensor),  in execute()
277                                  context->getOutputShape(kOutputTensor));  in execute()
281                                 context->getOutputBuffer<uint8_t>(kOutputTensor),  in execute()
282                                 context->getOutputShape(kOutputTensor));  in execute()
286                                       context->getOutputBuffer<int8_t>(kOutputTensor),  in execute()
287                                       context->getOutputShape(kOutputTensor));  in execute()