From 3e6676da92c228196c6cc7e6fe4e8891ac1b387c Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Thu, 14 May 2020 13:41:48 +0100 Subject: IVGCVSW-4669 Set destination tensorInfo in MoveAllConnections() Signed-off-by: Finn Williams Change-Id: I3563209dcb3db1b40cf2db3855adc631b5e323be --- src/armnn/optimizations/FoldPadIntoConvolution2d.hpp | 2 -- 1 file changed, 2 deletions(-) (limited to 'src/armnn/optimizations/FoldPadIntoConvolution2d.hpp') diff --git a/src/armnn/optimizations/FoldPadIntoConvolution2d.hpp b/src/armnn/optimizations/FoldPadIntoConvolution2d.hpp index 66fffbb280..5def6dfdd2 100644 --- a/src/armnn/optimizations/FoldPadIntoConvolution2d.hpp +++ b/src/armnn/optimizations/FoldPadIntoConvolution2d.hpp @@ -30,7 +30,6 @@ public: Convolution2dLayer* convolution2dLayer = PolymorphicDowncast(&child); OutputSlot* parentOut = base.GetInputSlot(0).GetConnectedOutputSlot(); - const TensorInfo& outInfo = child.GetOutputHandler().GetTensorInfo(); const std::string name = std::string("folded-") + base.GetName() + std::string("-into-") + child.GetName(); Convolution2dDescriptor descriptor = convolution2dLayer->GetParameters(); @@ -59,7 +58,6 @@ public: auto& newConv2dLayer = *graph.InsertNewLayer(base.GetInputSlot(0), descriptor, name.c_str()); - newConv2dLayer.GetOutputHandler().SetTensorInfo(outInfo); // Copy weights and bias to the new convolution layer ARMNN_ASSERT_MSG(convolution2dLayer->m_Weight != nullptr, -- cgit v1.2.1