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/PermuteAndBatchToSpaceAsDepthToSpace.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/PermuteAndBatchToSpaceAsDepthToSpace.hpp')
-rw-r--r-- | src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.hpp | 1 |
1 files changed, 0 insertions, 1 deletions
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<DepthToSpaceLayer>(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. |