Lines Matching refs:getOutputShape
410 auto outputShape = context->getOutputShape(kOutputTensor); in prepare()
435 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu()
441 context->getOutputShape(kOutputTensor)); in executeRelu()
446 context->getOutputShape(kOutputTensor)); in executeRelu()
451 context->getOutputShape(kOutputTensor)); in executeRelu()
456 context->getOutputShape(kOutputTensor)); in executeRelu()
464 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu1()
470 context->getOutputShape(kOutputTensor)); in executeRelu1()
475 context->getOutputShape(kOutputTensor)); in executeRelu1()
480 context->getOutputShape(kOutputTensor)); in executeRelu1()
485 context->getOutputShape(kOutputTensor)); in executeRelu1()
493 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu6()
499 context->getOutputShape(kOutputTensor)); in executeRelu6()
504 context->getOutputShape(kOutputTensor)); in executeRelu6()
509 context->getOutputShape(kOutputTensor)); in executeRelu6()
514 context->getOutputShape(kOutputTensor)); in executeRelu6()
522 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeLogistic()
528 context->getOutputShape(kOutputTensor)); in executeLogistic()
533 context->getOutputShape(kOutputTensor)); in executeLogistic()
538 context->getOutputShape(kOutputTensor)); in executeLogistic()
543 context->getOutputShape(kOutputTensor)); in executeLogistic()
551 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeTanh()
557 context->getOutputShape(kOutputTensor)); in executeTanh()
562 context->getOutputShape(kOutputTensor)); in executeTanh()
567 context->getOutputShape(kOutputTensor)); in executeTanh()
572 context->getOutputShape(kOutputTensor)); in executeTanh()
580 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeHardSwish()
584 const Shape& outputShape = context->getOutputShape(kOutputTensor); in executeHardSwish()
598 convertShapeToTflshape(context->getOutputShape(kOutputTensor)), in executeHardSwish()
606 context->getOutputShape(kOutputTensor)); in executeHardSwish()
611 context->getOutputShape(kOutputTensor)); in executeHardSwish()