• Home
  • Raw
  • Download

Lines Matching refs:kOutputTensor

35 constexpr uint32_t kOutputTensor = 0;  variable
251 return context->setOutputShape(kOutputTensor, output); in prepare()
256 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu()
261 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu()
262 context->getOutputShape(kOutputTensor)); in executeRelu()
266 context->getOutputBuffer<float>(kOutputTensor), in executeRelu()
267 context->getOutputShape(kOutputTensor)); in executeRelu()
271 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu()
272 context->getOutputShape(kOutputTensor)); in executeRelu()
280 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu1()
285 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu1()
286 context->getOutputShape(kOutputTensor)); in executeRelu1()
290 context->getOutputBuffer<float>(kOutputTensor), in executeRelu1()
291 context->getOutputShape(kOutputTensor)); in executeRelu1()
295 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu1()
296 context->getOutputShape(kOutputTensor)); in executeRelu1()
304 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu6()
309 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu6()
310 context->getOutputShape(kOutputTensor)); in executeRelu6()
314 context->getOutputBuffer<float>(kOutputTensor), in executeRelu6()
315 context->getOutputShape(kOutputTensor)); in executeRelu6()
319 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu6()
320 context->getOutputShape(kOutputTensor)); in executeRelu6()
328 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeLogistic()
333 context->getOutputBuffer<_Float16>(kOutputTensor), in executeLogistic()
334 context->getOutputShape(kOutputTensor)); in executeLogistic()
338 context->getOutputBuffer<float>(kOutputTensor), in executeLogistic()
339 context->getOutputShape(kOutputTensor)); in executeLogistic()
343 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeLogistic()
344 context->getOutputShape(kOutputTensor)); in executeLogistic()
352 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeTanh()
357 context->getOutputBuffer<_Float16>(kOutputTensor), in executeTanh()
358 context->getOutputShape(kOutputTensor)); in executeTanh()
362 context->getOutputBuffer<float>(kOutputTensor), in executeTanh()
363 context->getOutputShape(kOutputTensor)); in executeTanh()
367 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeTanh()
368 context->getOutputShape(kOutputTensor)); in executeTanh()