From 564c13dc098eb9353ac15e2609712ab8db9bf350 Mon Sep 17 00:00:00 2001 From: Idriss Chaouch Date: Fri, 1 Sep 2023 17:58:38 +0100 Subject: IVGCVSW-7525 Add broadcast_to to TFLite Parser * Changing the optimizer * Changing EndToEnd Tests Signed-off-by: Idriss Chaouch Signed-off-by: Narumol Prangnawarat Change-Id: Ib581794280322a39cfc5ea3c4e6a6398cf723d5e --- src/armnn/layers/BroadcastToLayer.cpp | 1 - src/armnn/optimizations/DeleteBroadcastTo.hpp | 3 +++ 2 files changed, 3 insertions(+), 1 deletion(-) (limited to 'src/armnn') diff --git a/src/armnn/layers/BroadcastToLayer.cpp b/src/armnn/layers/BroadcastToLayer.cpp index 252aa46de0..26a90eefed 100644 --- a/src/armnn/layers/BroadcastToLayer.cpp +++ b/src/armnn/layers/BroadcastToLayer.cpp @@ -6,7 +6,6 @@ #include "BroadcastToLayer.hpp" #include "LayerCloneBase.hpp" - #include #include #include 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); } } } -- cgit v1.2.1