diff options
author | Idriss Chaouch <idriss.chaouch@arm.com> | 2023-09-08 11:18:16 +0100 |
---|---|---|
committer | Idriss Chaouch <idriss.chaouch@arm.com> | 2023-09-12 11:18:53 +0100 |
commit | cbf79298f73310fe4ca5d760ded73575e4bf8fad (patch) | |
tree | 95d1c85c20a6af48427ea50721d65ea6e72cd10d /delegate/opaque/src/armnn_delegate.cpp | |
parent | cd0874c5e404f8676d988f7d241aa1f88a037dd2 (diff) | |
download | armnn-cbf79298f73310fe4ca5d760ded73575e4bf8fad.tar.gz |
IVGCVSW-8037 Add BROADCAST_TO to tflite classic and opaque delegate.
Signed-off-by: Idriss Chaouch <idriss.chaouch@arm.com>
Change-Id: Ibc145d0ea1ac9414b6a68b5b547bf2ea2852fd36
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 bad1abaa59..08b1504efb 100644 --- a/delegate/opaque/src/armnn_delegate.cpp +++ b/delegate/opaque/src/armnn_delegate.cpp @@ -10,6 +10,7 @@ #include "ArgMinMax.hpp" #include "BatchMatMul.hpp" #include "BatchSpace.hpp" +#include "BroadcastTo.hpp" #include "Comparison.hpp" #include "Convolution.hpp" #include "Control.hpp" @@ -654,6 +655,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, kTfLiteBuiltinBatchMatmul); + case kTfLiteBuiltinBroadcastTo: + return VisitBroadcastToOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinBroadcastTo); case kTfLiteBuiltinBatchToSpaceNd: return VisitBatchToSpaceNdOperator(delegateData, tfLiteContext, |