diff options
author | Idriss Chaouch <idriss.chaouch@arm.com> | 2023-09-01 17:58:38 +0100 |
---|---|---|
committer | idriss.chaouch <idriss.chaouch@arm.com> | 2023-09-08 08:32:43 +0000 |
commit | 564c13dc098eb9353ac15e2609712ab8db9bf350 (patch) | |
tree | 6cb52e904e3cd001d650a6386b1105ee21b08847 /src/armnn/optimizations | |
parent | 04e3eb5d339c3778f26c69651bf1464c8ab5331d (diff) | |
download | armnn-564c13dc098eb9353ac15e2609712ab8db9bf350.tar.gz |
IVGCVSW-7525 Add broadcast_to to TFLite Parser
* Changing the optimizer
* Changing EndToEnd Tests
Signed-off-by: Idriss Chaouch <idriss.chaouch@arm.com>
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ib581794280322a39cfc5ea3c4e6a6398cf723d5e
Diffstat (limited to 'src/armnn/optimizations')
-rw-r--r-- | src/armnn/optimizations/DeleteBroadcastTo.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnn/optimizations/DeleteBroadcastTo.hpp b/src/armnn/optimizations/DeleteBroadcastTo.hpp index 9ea20907df..38396c1a9c 100644 --- a/src/armnn/optimizations/DeleteBroadcastTo.hpp +++ b/src/armnn/optimizations/DeleteBroadcastTo.hpp @@ -20,11 +20,14 @@ public: { if(layer.GetType() == LayerType::BroadcastTo) { + TensorInfo info = layer.GetOutputSlot(0).GetTensorInfo(); Layer& next = layer.GetOutputSlot(0).GetConnection(0)->GetOwningLayer(); if (next.GetType() == LayerType::ElementwiseBinary) { Layer& connectedLayer = layer.GetInputSlots()[0].GetConnectedOutputSlot()->GetOwningLayer(); + auto tensorInfo = connectedLayer.GetOutputSlot().GetTensorInfo(); layer.GetOutputSlot().MoveAllConnections(connectedLayer.GetOutputSlot()); + connectedLayer.GetOutputSlot().GetOutputHandler().SetTensorInfo(tensorInfo); } } } |