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/PermuteAsReshape.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/PermuteAsReshape.hpp')
-rw-r--r-- | src/armnn/optimizations/PermuteAsReshape.hpp | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/src/armnn/optimizations/PermuteAsReshape.hpp b/src/armnn/optimizations/PermuteAsReshape.hpp index 68fa667085..7878ecfb93 100644 --- a/src/armnn/optimizations/PermuteAsReshape.hpp +++ b/src/armnn/optimizations/PermuteAsReshape.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<ReshapeLayer>(permute.GetInputSlot(0), descriptor, name.c_str()); - reshape->GetOutputHandler().SetTensorInfo(outInfo); // Bypass permute. It will be deleted since it's left unconnected. permute.GetOutputSlot().MoveAllConnections(reshape->GetOutputSlot()); |