Lines Matching refs:context
441 bool prepare(IOperationExecutionContext* context) { in prepare() argument
442 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
443 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
444 Shape output = context->getOutputShape(kOutputTensor); in prepare()
448 return context->setOutputShape(kOutputTensor, output); in prepare()
451 bool executeAdd(IOperationExecutionContext* context) { in executeAdd() argument
453 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
454 switch (context->getInputType(kInputTensor1)) { in executeAdd()
456 return addFloat16(context->getInputBuffer<_Float16>(kInputTensor1), in executeAdd()
457 context->getInputShape(kInputTensor1), in executeAdd()
458 context->getInputBuffer<_Float16>(kInputTensor2), in executeAdd()
459 context->getInputShape(kInputTensor2), in executeAdd()
460 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
461 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
462 context->getOutputShape(kOutputTensor)); in executeAdd()
464 return addFloat32(context->getInputBuffer<float>(kInputTensor1), in executeAdd()
465 context->getInputShape(kInputTensor1), in executeAdd()
466 context->getInputBuffer<float>(kInputTensor2), in executeAdd()
467 context->getInputShape(kInputTensor2), in executeAdd()
468 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
469 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
470 context->getOutputShape(kOutputTensor)); in executeAdd()
472 return addQuant8(context->getInputBuffer<uint8_t>(kInputTensor1), in executeAdd()
473 context->getInputShape(kInputTensor1), in executeAdd()
474 context->getInputBuffer<uint8_t>(kInputTensor2), in executeAdd()
475 context->getInputShape(kInputTensor2), in executeAdd()
476 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
477 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
478 context->getOutputShape(kOutputTensor)); in executeAdd()
480 return addQuant8(context->getInputBuffer<int8_t>(kInputTensor1), in executeAdd()
481 context->getInputShape(kInputTensor1), in executeAdd()
482 context->getInputBuffer<int8_t>(kInputTensor2), in executeAdd()
483 context->getInputShape(kInputTensor2), in executeAdd()
484 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
485 context->getOutputBuffer<int8_t>(kOutputTensor), in executeAdd()
486 context->getOutputShape(kOutputTensor)); in executeAdd()
488 return executeInt32(context->getInputBuffer<int32_t>(kInputTensor1), in executeAdd()
489 context->getInputShape(kInputTensor1), in executeAdd()
490 context->getInputBuffer<int32_t>(kInputTensor2), in executeAdd()
491 context->getInputShape(kInputTensor2), in executeAdd()
492 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
493 context->getOutputBuffer<int32_t>(kOutputTensor), in executeAdd()
494 context->getOutputShape(kOutputTensor), in executeAdd()
501 bool executeMul(IOperationExecutionContext* context) { in executeMul() argument
503 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
504 switch (context->getInputType(kInputTensor1)) { in executeMul()
506 return mulFloat16(context->getInputBuffer<_Float16>(kInputTensor1), in executeMul()
507 context->getInputShape(kInputTensor1), in executeMul()
508 context->getInputBuffer<_Float16>(kInputTensor2), in executeMul()
509 context->getInputShape(kInputTensor2), in executeMul()
510 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
511 context->getOutputBuffer<_Float16>(kOutputTensor), in executeMul()
512 context->getOutputShape(kOutputTensor)); in executeMul()
514 return mulFloat32(context->getInputBuffer<float>(kInputTensor1), in executeMul()
515 context->getInputShape(kInputTensor1), in executeMul()
516 context->getInputBuffer<float>(kInputTensor2), in executeMul()
517 context->getInputShape(kInputTensor2), in executeMul()
518 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
519 context->getOutputBuffer<float>(kOutputTensor), in executeMul()
520 context->getOutputShape(kOutputTensor)); in executeMul()
522 return mulQuant8(context->getInputBuffer<uint8_t>(kInputTensor1), in executeMul()
523 context->getInputShape(kInputTensor1), in executeMul()
524 context->getInputBuffer<uint8_t>(kInputTensor2), in executeMul()
525 context->getInputShape(kInputTensor2), in executeMul()
526 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
527 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeMul()
528 context->getOutputShape(kOutputTensor)); in executeMul()
530 return mulQuant8(context->getInputBuffer<int8_t>(kInputTensor1), in executeMul()
531 context->getInputShape(kInputTensor1), in executeMul()
532 context->getInputBuffer<int8_t>(kInputTensor2), in executeMul()
533 context->getInputShape(kInputTensor2), in executeMul()
534 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
535 context->getOutputBuffer<int8_t>(kOutputTensor), in executeMul()
536 context->getOutputShape(kOutputTensor)); in executeMul()
538 return executeInt32(context->getInputBuffer<int32_t>(kInputTensor1), in executeMul()
539 context->getInputShape(kInputTensor1), in executeMul()
540 context->getInputBuffer<int32_t>(kInputTensor2), in executeMul()
541 context->getInputShape(kInputTensor2), in executeMul()
542 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
543 context->getOutputBuffer<int32_t>(kOutputTensor), in executeMul()
544 context->getOutputShape(kOutputTensor), in executeMul()
551 bool executeSub(IOperationExecutionContext* context) { in executeSub() argument
553 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeSub()
554 switch (context->getInputType(kInputTensor1)) { in executeSub()
556 return subFloat16(context->getInputBuffer<_Float16>(kInputTensor1), in executeSub()
557 context->getInputShape(kInputTensor1), in executeSub()
558 context->getInputBuffer<_Float16>(kInputTensor2), in executeSub()
559 context->getInputShape(kInputTensor2), in executeSub()
560 context->getInputValue<int32_t>(kActivationScalar), in executeSub()
561 context->getOutputBuffer<_Float16>(kOutputTensor), in executeSub()
562 context->getOutputShape(kOutputTensor)); in executeSub()
564 return subFloat32(context->getInputBuffer<float>(kInputTensor1), in executeSub()
565 context->getInputShape(kInputTensor1), in executeSub()
566 context->getInputBuffer<float>(kInputTensor2), in executeSub()
567 context->getInputShape(kInputTensor2), in executeSub()
568 context->getInputValue<int32_t>(kActivationScalar), in executeSub()
569 context->getOutputBuffer<float>(kOutputTensor), in executeSub()
570 context->getOutputShape(kOutputTensor)); in executeSub()
572 return subQuant8(context->getInputBuffer<uint8_t>(kInputTensor1), in executeSub()
573 context->getInputShape(kInputTensor1), in executeSub()
574 context->getInputBuffer<uint8_t>(kInputTensor2), in executeSub()
575 context->getInputShape(kInputTensor2), in executeSub()
576 context->getInputValue<int32_t>(kActivationScalar), in executeSub()
577 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeSub()
578 context->getOutputShape(kOutputTensor)); in executeSub()
580 return subQuant8(context->getInputBuffer<int8_t>(kInputTensor1), in executeSub()
581 context->getInputShape(kInputTensor1), in executeSub()
582 context->getInputBuffer<int8_t>(kInputTensor2), in executeSub()
583 context->getInputShape(kInputTensor2), in executeSub()
584 context->getInputValue<int32_t>(kActivationScalar), in executeSub()
585 context->getOutputBuffer<int8_t>(kOutputTensor), in executeSub()
586 context->getOutputShape(kOutputTensor)); in executeSub()
588 return executeInt32(context->getInputBuffer<int32_t>(kInputTensor1), in executeSub()
589 context->getInputShape(kInputTensor1), in executeSub()
590 context->getInputBuffer<int32_t>(kInputTensor2), in executeSub()
591 context->getInputShape(kInputTensor2), in executeSub()
592 context->getInputValue<int32_t>(kActivationScalar), in executeSub()
593 context->getOutputBuffer<int32_t>(kOutputTensor), in executeSub()
594 context->getOutputShape(kOutputTensor), in executeSub()
601 bool executeDiv(IOperationExecutionContext* context) { in executeDiv() argument
603 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeDiv()
604 switch (context->getInputType(kInputTensor1)) { in executeDiv()
606 return divFloat16(context->getInputBuffer<_Float16>(kInputTensor1), in executeDiv()
607 context->getInputShape(kInputTensor1), in executeDiv()
608 context->getInputBuffer<_Float16>(kInputTensor2), in executeDiv()
609 context->getInputShape(kInputTensor2), in executeDiv()
610 context->getInputValue<int32_t>(kActivationScalar), in executeDiv()
611 context->getOutputBuffer<_Float16>(kOutputTensor), in executeDiv()
612 context->getOutputShape(kOutputTensor)); in executeDiv()
614 return divFloat32(context->getInputBuffer<float>(kInputTensor1), in executeDiv()
615 context->getInputShape(kInputTensor1), in executeDiv()
616 context->getInputBuffer<float>(kInputTensor2), in executeDiv()
617 context->getInputShape(kInputTensor2), in executeDiv()
618 context->getInputValue<int32_t>(kActivationScalar), in executeDiv()
619 context->getOutputBuffer<float>(kOutputTensor), in executeDiv()
620 context->getOutputShape(kOutputTensor)); in executeDiv()
622 return executeInt32(context->getInputBuffer<int32_t>(kInputTensor1), in executeDiv()
623 context->getInputShape(kInputTensor1), in executeDiv()
624 context->getInputBuffer<int32_t>(kInputTensor2), in executeDiv()
625 context->getInputShape(kInputTensor2), in executeDiv()
626 context->getInputValue<int32_t>(kActivationScalar), in executeDiv()
627 context->getOutputBuffer<int32_t>(kOutputTensor), in executeDiv()
628 context->getOutputShape(kOutputTensor), [](int32_t a, int32_t b) { in executeDiv()