diff options
Diffstat (limited to 'delegate/classic/src/armnn_delegate.cpp')
| -rw-r--r-- | delegate/classic/src/armnn_delegate.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/delegate/classic/src/armnn_delegate.cpp b/delegate/classic/src/armnn_delegate.cpp index 05bf9b2ee7..52621ee6c1 100644 --- a/delegate/classic/src/armnn_delegate.cpp +++ b/delegate/classic/src/armnn_delegate.cpp @@ -34,6 +34,7 @@ #include "Resize.hpp" #include "ReverseV2.hpp" #include "Round.hpp" +#include "ScatterNd.hpp" #include "Shape.hpp" #include "Slice.hpp" #include "StridedSlice.hpp" @@ -1070,6 +1071,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, kTfLiteBuiltinTransposeConv); + case kTfLiteBuiltinScatterNd: + return VisitScatterNdOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinScatterNd); case kTfLiteBuiltinSoftmax: return VisitSoftmaxOperator(delegateData, tfLiteContext, |