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/classic/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/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 de2aa0c632..c428d46d87 100644 --- a/delegate/classic/src/armnn_delegate.cpp +++ b/delegate/classic/src/armnn_delegate.cpp @@ -11,6 +11,7 @@ #include "ArgMinMax.hpp" #include "BatchMatMul.hpp" #include "BatchSpace.hpp" +#include "BroadcastTo.hpp" #include "Comparison.hpp" #include "Convolution.hpp" #include "Control.hpp" @@ -603,6 +604,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, kTfLiteBuiltinBatchToSpaceNd); + case kTfLiteBuiltinBroadcastTo: + return VisitBroadcastToOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinBroadcastTo); case kTfLiteBuiltinCast: return VisitCastOperator(delegateData, tfLiteContext, |