Lines Matching refs:kOutputTensor
50 constexpr uint32_t kOutputTensor = 0; variable
455 Shape output = context->getOutputShape(kOutputTensor); in validate()
482 Shape output = context->getOutputShape(kOutputTensor); in prepare()
486 return context->setOutputShape(kOutputTensor, output); in prepare()
491 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
499 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
500 context->getOutputShape(kOutputTensor)); in executeAdd()
507 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
508 context->getOutputShape(kOutputTensor)); in executeAdd()
515 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
516 context->getOutputShape(kOutputTensor)); in executeAdd()
523 context->getOutputBuffer<int8_t>(kOutputTensor), in executeAdd()
524 context->getOutputShape(kOutputTensor)); in executeAdd()
531 context->getOutputBuffer<int32_t>(kOutputTensor), in executeAdd()
532 context->getOutputShape(kOutputTensor), in executeAdd()
541 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
549 context->getOutputBuffer<_Float16>(kOutputTensor), in executeMul()
550 context->getOutputShape(kOutputTensor)); in executeMul()
557 context->getOutputBuffer<float>(kOutputTensor), in executeMul()
558 context->getOutputShape(kOutputTensor)); in executeMul()
565 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeMul()
566 context->getOutputShape(kOutputTensor)); in executeMul()
573 context->getOutputBuffer<int8_t>(kOutputTensor), in executeMul()
574 context->getOutputShape(kOutputTensor)); in executeMul()
581 context->getOutputBuffer<int32_t>(kOutputTensor), in executeMul()
582 context->getOutputShape(kOutputTensor), in executeMul()
591 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeSub()
599 context->getOutputBuffer<_Float16>(kOutputTensor), in executeSub()
600 context->getOutputShape(kOutputTensor)); in executeSub()
607 context->getOutputBuffer<float>(kOutputTensor), in executeSub()
608 context->getOutputShape(kOutputTensor)); in executeSub()
615 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeSub()
616 context->getOutputShape(kOutputTensor)); in executeSub()
623 context->getOutputBuffer<int8_t>(kOutputTensor), in executeSub()
624 context->getOutputShape(kOutputTensor)); in executeSub()
631 context->getOutputBuffer<int32_t>(kOutputTensor), in executeSub()
632 context->getOutputShape(kOutputTensor), in executeSub()
641 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeDiv()
649 context->getOutputBuffer<_Float16>(kOutputTensor), in executeDiv()
650 context->getOutputShape(kOutputTensor)); in executeDiv()
657 context->getOutputBuffer<float>(kOutputTensor), in executeDiv()
658 context->getOutputShape(kOutputTensor)); in executeDiv()
665 context->getOutputBuffer<int32_t>(kOutputTensor), in executeDiv()
666 context->getOutputShape(kOutputTensor), [](int32_t a, int32_t b) { in executeDiv()