Lines Matching refs:getOutputShape
455 Shape output = context->getOutputShape(kOutputTensor); in validate()
482 Shape output = context->getOutputShape(kOutputTensor); in prepare()
491 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
500 context->getOutputShape(kOutputTensor)); in executeAdd()
508 context->getOutputShape(kOutputTensor)); in executeAdd()
516 context->getOutputShape(kOutputTensor)); in executeAdd()
524 context->getOutputShape(kOutputTensor)); in executeAdd()
532 context->getOutputShape(kOutputTensor), in executeAdd()
541 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
550 context->getOutputShape(kOutputTensor)); in executeMul()
558 context->getOutputShape(kOutputTensor)); in executeMul()
566 context->getOutputShape(kOutputTensor)); in executeMul()
574 context->getOutputShape(kOutputTensor)); in executeMul()
582 context->getOutputShape(kOutputTensor), in executeMul()
591 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeSub()
600 context->getOutputShape(kOutputTensor)); in executeSub()
608 context->getOutputShape(kOutputTensor)); in executeSub()
616 context->getOutputShape(kOutputTensor)); in executeSub()
624 context->getOutputShape(kOutputTensor)); in executeSub()
632 context->getOutputShape(kOutputTensor), in executeSub()
641 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeDiv()
650 context->getOutputShape(kOutputTensor)); in executeDiv()
658 context->getOutputShape(kOutputTensor)); in executeDiv()
666 context->getOutputShape(kOutputTensor), [](int32_t a, int32_t b) { in executeDiv()