| 704 | } |
| 705 | |
| 706 | TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, |
| 707 | TfLiteOpaqueContext* tfLiteContext, |
| 708 | TfLiteRegistrationExternal* tfLiteRegistration, |
| 709 | TfLiteOpaqueNode* tfLiteNode, |
| 710 | int nodeIndex) |
| 711 | { |
| 712 | switch (TfLiteRegistrationExternalGetBuiltInCode(tfLiteRegistration)) |
| 713 | { |
| 714 | case kTfLiteBuiltinAbs: |
| 715 | return VisitElementwiseUnaryOperator(delegateData, |
| 716 | tfLiteContext, |
| 717 | tfLiteNode, |
| 718 | nodeIndex, |
| 719 | kTfLiteBuiltinAbs, |
| 720 | armnn::UnaryOperation::Abs); |
| 721 | case kTfLiteBuiltinAdd: |
| 722 | return VisitElementwiseBinaryOperator(delegateData, |
| 723 | tfLiteContext, |
| 724 | tfLiteNode, |
| 725 | nodeIndex, |
| 726 | kTfLiteBuiltinAdd); |
| 727 | case kTfLiteBuiltinArgMax: |
| 728 | return VisitArgMinMaxOperator(delegateData, |
| 729 | tfLiteContext, |
| 730 | tfLiteNode, |
| 731 | nodeIndex, |
| 732 | kTfLiteBuiltinArgMax); |
| 733 | case kTfLiteBuiltinArgMin: |
| 734 | return VisitArgMinMaxOperator(delegateData, |
| 735 | tfLiteContext, |
| 736 | tfLiteNode, |
| 737 | nodeIndex, |
| 738 | kTfLiteBuiltinArgMin); |
| 739 | case kTfLiteBuiltinAveragePool2d: |
| 740 | return VisitPooling2dOperator(delegateData, |
| 741 | tfLiteContext, |
| 742 | tfLiteNode, |
| 743 | nodeIndex, |
| 744 | kTfLiteBuiltinAveragePool2d); |
| 745 | case kTfLiteBuiltinBatchMatmul: |
| 746 | return VisitBatchMatMulOperator(delegateData, |
| 747 | tfLiteContext, |
| 748 | tfLiteNode, |
| 749 | nodeIndex, |
| 750 | kTfLiteBuiltinBatchMatmul); |
| 751 | case kTfLiteBuiltinBroadcastTo: |
| 752 | return VisitBroadcastToOperator(delegateData, |
| 753 | tfLiteContext, |
| 754 | tfLiteNode, |
| 755 | nodeIndex, |
| 756 | kTfLiteBuiltinBroadcastTo); |
| 757 | case kTfLiteBuiltinBatchToSpaceNd: |
| 758 | return VisitBatchToSpaceNdOperator(delegateData, |
| 759 | tfLiteContext, |
| 760 | tfLiteNode, |
| 761 | nodeIndex, |
| 762 | kTfLiteBuiltinBatchToSpaceNd); |
| 763 | case kTfLiteBuiltinCast: |
nothing calls this directly
no test coverage detected