diff options
author | Finn Williams <Finn.Williams@arm.com> | 2020-05-14 13:41:48 +0100 |
---|---|---|
committer | finn.williams <finn.williams@arm.com> | 2020-05-19 14:14:30 +0000 |
commit | 3e6676da92c228196c6cc7e6fe4e8891ac1b387c (patch) | |
tree | d9d0683b2a2f1d1d06c51a45f3528e3789c1b1d1 /src/armnn/optimizations/FoldPadIntoConvolution2d.hpp | |
parent | c4532d10b88ebe3d82b6b9f109f3cae035b5a4df (diff) | |
download | armnn-3e6676da92c228196c6cc7e6fe4e8891ac1b387c.tar.gz |
IVGCVSW-4669 Set destination tensorInfo in MoveAllConnections()
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I3563209dcb3db1b40cf2db3855adc631b5e323be
Diffstat (limited to 'src/armnn/optimizations/FoldPadIntoConvolution2d.hpp')
-rw-r--r-- | src/armnn/optimizations/FoldPadIntoConvolution2d.hpp | 2 |
1 files changed, 0 insertions, 2 deletions
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<Convolution2dLayer*>(&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<Convolution2dLayer>(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, |