• Home
  • Raw
  • Download

Lines Matching refs:kOutputTensor

52 constexpr uint32_t kOutputTensor = 0;  variable
459 Shape output = context->getOutputShape(kOutputTensor); in validate()
488 Shape output = context->getOutputShape(kOutputTensor); in prepare()
492 return context->setOutputShape(kOutputTensor, output); in prepare()
497 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
505 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
506 context->getOutputShape(kOutputTensor)); in executeAdd()
513 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
514 context->getOutputShape(kOutputTensor)); in executeAdd()
521 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
522 context->getOutputShape(kOutputTensor)); in executeAdd()
529 context->getOutputBuffer<int8_t>(kOutputTensor), in executeAdd()
530 context->getOutputShape(kOutputTensor)); in executeAdd()
537 context->getOutputBuffer<int32_t>(kOutputTensor), in executeAdd()
538 context->getOutputShape(kOutputTensor), in executeAdd()
547 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
555 context->getOutputBuffer<_Float16>(kOutputTensor), in executeMul()
556 context->getOutputShape(kOutputTensor)); in executeMul()
563 context->getOutputBuffer<float>(kOutputTensor), in executeMul()
564 context->getOutputShape(kOutputTensor)); in executeMul()
571 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeMul()
572 context->getOutputShape(kOutputTensor)); in executeMul()
579 context->getOutputBuffer<int8_t>(kOutputTensor), in executeMul()
580 context->getOutputShape(kOutputTensor)); in executeMul()
587 context->getOutputBuffer<int32_t>(kOutputTensor), in executeMul()
588 context->getOutputShape(kOutputTensor), in executeMul()
597 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeSub()
605 context->getOutputBuffer<_Float16>(kOutputTensor), in executeSub()
606 context->getOutputShape(kOutputTensor)); in executeSub()
613 context->getOutputBuffer<float>(kOutputTensor), in executeSub()
614 context->getOutputShape(kOutputTensor)); in executeSub()
621 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeSub()
622 context->getOutputShape(kOutputTensor)); in executeSub()
629 context->getOutputBuffer<int8_t>(kOutputTensor), in executeSub()
630 context->getOutputShape(kOutputTensor)); in executeSub()
637 context->getOutputBuffer<int32_t>(kOutputTensor), in executeSub()
638 context->getOutputShape(kOutputTensor), in executeSub()
647 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeDiv()
655 context->getOutputBuffer<_Float16>(kOutputTensor), in executeDiv()
656 context->getOutputShape(kOutputTensor)); in executeDiv()
663 context->getOutputBuffer<float>(kOutputTensor), in executeDiv()
664 context->getOutputShape(kOutputTensor)); in executeDiv()
671 context->getOutputBuffer<int32_t>(kOutputTensor), in executeDiv()
672 context->getOutputShape(kOutputTensor), [](int32_t a, int32_t b) { in executeDiv()