diff options
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()); } } |