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/PermuteAndBatchToSpaceAsDepthToSpace.hpp | 1 - 1 file changed, 1 deletion(-) (limited to 'src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.hpp') diff --git a/src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.hpp b/src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.hpp index ea4de9df6f..d94eb7ca0d 100644 --- a/src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.hpp +++ b/src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.hpp @@ -90,7 +90,6 @@ public: auto& depthToSpace = *graph.InsertNewLayer(base.GetInputSlot(0), depthToSpaceDesc, name.c_str()); - depthToSpace.GetOutputHandler().SetTensorInfo(outputInfo); // Moves connections from child output to new layer. // Child layer will be removed as it's left unconnected. -- cgit v1.2.1