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/OptimizeConsecutiveReshapes.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/OptimizeConsecutiveReshapes.hpp')
-rw-r--r-- | src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp b/src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp index 39bfe6e936..546b37d525 100644 --- a/src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp +++ b/src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp @@ -43,8 +43,6 @@ public: const ReshapeDescriptor descriptor{outInfo.GetShape()}; auto& newReshape = *graph.InsertNewLayer<ReshapeLayer>(base.GetInputSlot(0), descriptor, name.c_str()); - // Sets tensor info for new layer. - newReshape.GetOutputHandler().SetTensorInfo(outInfo); // Parent is now the new layer. parentOut = &newReshape.GetOutputSlot(); } |