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/Layer.cpp | |
| 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/Layer.cpp')
| -rw-r--r-- | src/armnn/Layer.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnn/Layer.cpp b/src/armnn/Layer.cpp index 024a18862d..b7a6fd96e3 100644 --- a/src/armnn/Layer.cpp +++ b/src/armnn/Layer.cpp @@ -119,6 +119,7 @@ void OutputSlot::MoveAllConnections(OutputSlot& destination) InputSlot& connection = *GetConnection(0); Disconnect(connection); destination.Connect(connection); + destination.GetOutputHandler().SetTensorInfo(GetOutputHandler().GetTensorInfo()); } } |