• Home
  • Raw
  • Download

Lines Matching refs:kOutputTensor

41 constexpr uint32_t kOutputTensor = 0;  variable
367 Shape output = context->getOutputShape(kOutputTensor); in validate()
385 Shape output = context->getOutputShape(kOutputTensor); in prepare()
389 return context->setOutputShape(kOutputTensor, output); in prepare()
394 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
402 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
403 context->getOutputShape(kOutputTensor)); in executeAdd()
410 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
411 context->getOutputShape(kOutputTensor)); in executeAdd()
418 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
419 context->getOutputShape(kOutputTensor)); in executeAdd()
427 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
435 context->getOutputBuffer<_Float16>(kOutputTensor), in executeMul()
436 context->getOutputShape(kOutputTensor)); in executeMul()
443 context->getOutputBuffer<float>(kOutputTensor), in executeMul()
444 context->getOutputShape(kOutputTensor)); in executeMul()
451 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeMul()
452 context->getOutputShape(kOutputTensor)); in executeMul()
460 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeSub()
468 context->getOutputBuffer<_Float16>(kOutputTensor), in executeSub()
469 context->getOutputShape(kOutputTensor)); in executeSub()
476 context->getOutputBuffer<float>(kOutputTensor), in executeSub()
477 context->getOutputShape(kOutputTensor)); in executeSub()
484 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeSub()
485 context->getOutputShape(kOutputTensor)); in executeSub()
493 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeDiv()
501 context->getOutputBuffer<_Float16>(kOutputTensor), in executeDiv()
502 context->getOutputShape(kOutputTensor)); in executeDiv()
509 context->getOutputBuffer<float>(kOutputTensor), in executeDiv()
510 context->getOutputShape(kOutputTensor)); in executeDiv()