Lines Matching refs:kOutputTensor

42 constexpr uint32_t kOutputTensor = 0;  variable
125     Shape output = context->getOutputShape(kOutputTensor);  in prepare()
127     return context->setOutputShape(kOutputTensor, output);  in prepare()
141                     context->getOutputBuffer<_Float16>(kOutputTensor),  in execute()
142                     context->getOutputShape(kOutputTensor));  in execute()
152                     context->getOutputBuffer<float>(kOutputTensor),  in execute()
153                     context->getOutputShape(kOutputTensor));  in execute()
159                               context->getOutputBuffer<uint8_t>(kOutputTensor),  in execute()
160                               context->getOutputShape(kOutputTensor));  in execute()
167                               context->getOutputBuffer<int8_t>(kOutputTensor),  in execute()
168                               context->getOutputShape(kOutputTensor));  in execute()