diff options
Diffstat (limited to 'delegate/opaque/src/armnn_delegate.cpp')
| -rw-r--r-- | delegate/opaque/src/armnn_delegate.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/delegate/opaque/src/armnn_delegate.cpp b/delegate/opaque/src/armnn_delegate.cpp index 4ed0a78013..9e047f6a68 100644 --- a/delegate/opaque/src/armnn_delegate.cpp +++ b/delegate/opaque/src/armnn_delegate.cpp @@ -33,6 +33,7 @@ #include "Resize.hpp" #include "ReverseV2.hpp" #include "Round.hpp" +#include "ScatterNd.hpp" #include "Shape.hpp" #include "Slice.hpp" #include "StridedSlice.hpp" @@ -1154,6 +1155,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, nodeIndex, kTfLiteBuiltinRsqrt, armnn::UnaryOperation::Rsqrt); + case kTfLiteBuiltinScatterNd: + return VisitScatterNdOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinScatterNd); case kTfLiteBuiltinShape: return VisitShapeOperator(delegateData, tfLiteContext, |