Lines Matching refs:getOutputShape
367 Shape output = context->getOutputShape(kOutputTensor); in validate()
385 Shape output = context->getOutputShape(kOutputTensor); in prepare()
394 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
403 context->getOutputShape(kOutputTensor)); in executeAdd()
411 context->getOutputShape(kOutputTensor)); in executeAdd()
419 context->getOutputShape(kOutputTensor)); in executeAdd()
427 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
436 context->getOutputShape(kOutputTensor)); in executeMul()
444 context->getOutputShape(kOutputTensor)); in executeMul()
452 context->getOutputShape(kOutputTensor)); in executeMul()
460 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeSub()
469 context->getOutputShape(kOutputTensor)); in executeSub()
477 context->getOutputShape(kOutputTensor)); in executeSub()
485 context->getOutputShape(kOutputTensor)); in executeSub()
493 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeDiv()
502 context->getOutputShape(kOutputTensor)); in executeDiv()
510 context->getOutputShape(kOutputTensor)); in executeDiv()