Lines Matching refs:kOutputTensor
443 Shape output = context->getOutputShape(kOutputTensor); in prepare()
447 return context->setOutputShape(kOutputTensor, output); in prepare()
452 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
460 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
461 context->getOutputShape(kOutputTensor)); in executeAdd()
468 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
469 context->getOutputShape(kOutputTensor)); in executeAdd()
476 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
477 context->getOutputShape(kOutputTensor)); in executeAdd()
484 context->getOutputBuffer<int8_t>(kOutputTensor), in executeAdd()
485 context->getOutputShape(kOutputTensor)); in executeAdd()
492 context->getOutputBuffer<int32_t>(kOutputTensor), in executeAdd()
493 context->getOutputShape(kOutputTensor), in executeAdd()
502 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
510 context->getOutputBuffer<_Float16>(kOutputTensor), in executeMul()
511 context->getOutputShape(kOutputTensor)); in executeMul()
518 context->getOutputBuffer<float>(kOutputTensor), in executeMul()
519 context->getOutputShape(kOutputTensor)); in executeMul()
526 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeMul()
527 context->getOutputShape(kOutputTensor)); in executeMul()
534 context->getOutputBuffer<int8_t>(kOutputTensor), in executeMul()
535 context->getOutputShape(kOutputTensor)); in executeMul()
542 context->getOutputBuffer<int32_t>(kOutputTensor), in executeMul()
543 context->getOutputShape(kOutputTensor), in executeMul()
552 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeSub()
560 context->getOutputBuffer<_Float16>(kOutputTensor), in executeSub()
561 context->getOutputShape(kOutputTensor)); in executeSub()
568 context->getOutputBuffer<float>(kOutputTensor), in executeSub()
569 context->getOutputShape(kOutputTensor)); in executeSub()
576 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeSub()
577 context->getOutputShape(kOutputTensor)); in executeSub()
584 context->getOutputBuffer<int8_t>(kOutputTensor), in executeSub()
585 context->getOutputShape(kOutputTensor)); in executeSub()
592 context->getOutputBuffer<int32_t>(kOutputTensor), in executeSub()
593 context->getOutputShape(kOutputTensor), in executeSub()
602 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeDiv()
610 context->getOutputBuffer<_Float16>(kOutputTensor), in executeDiv()
611 context->getOutputShape(kOutputTensor)); in executeDiv()
618 context->getOutputBuffer<float>(kOutputTensor), in executeDiv()
619 context->getOutputShape(kOutputTensor)); in executeDiv()
626 context->getOutputBuffer<int32_t>(kOutputTensor), in executeDiv()
627 context->getOutputShape(kOutputTensor), [](int32_t a, int32_t b) { in executeDiv()