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/TransposeAsReshape.hpp | 1 - 1 file changed, 1 deletion(-) (limited to 'src/armnn/optimizations/TransposeAsReshape.hpp') diff --git a/src/armnn/optimizations/TransposeAsReshape.hpp b/src/armnn/optimizations/TransposeAsReshape.hpp index 4bb2f192f3..d6fb9bc316 100644 --- a/src/armnn/optimizations/TransposeAsReshape.hpp +++ b/src/armnn/optimizations/TransposeAsReshape.hpp @@ -25,7 +25,6 @@ public: const ReshapeDescriptor descriptor{outInfo.GetShape()}; // Inserts NewLayer so layers don't need to be re-sorted. auto reshape = graph.InsertNewLayer(transpose.GetInputSlot(0), descriptor, name.c_str()); - reshape->GetOutputHandler().SetTensorInfo(outInfo); // Bypass transpose. It will be deleted since it's left unconnected. transpose.GetOutputSlot().MoveAllConnections(reshape->GetOutputSlot()); -- cgit v1.2.1