diff options
author | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
---|---|---|
committer | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
commit | c577f2c6a3b4ddb6ba87a882723c53a248afbeba (patch) | |
tree | bd7d4c148df27f8be6649d313efb24f536b7cf34 /src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp | |
parent | 4c7098bfeab1ffe1cdc77f6c15548d3e73274746 (diff) | |
download | armnn-c577f2c6a3b4ddb6ba87a882723c53a248afbeba.tar.gz |
Release 18.08
Diffstat (limited to 'src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp')
-rw-r--r-- | src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp b/src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp index 9a926a57a4..935186d32e 100644 --- a/src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp +++ b/src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp @@ -31,19 +31,19 @@ public: if (inInfo.GetShape() != outInfo.GetShape()) { - // Insert equivalent reshape before base layer + // Inserts equivalent reshape before base layer. const std::string name = std::string("merged-") + base.GetName() + std::string("-with-") + child.GetName(); const ReshapeDescriptor descriptor{outInfo.GetShape()}; auto& newReshape = *graph.InsertNewLayer<ReshapeLayer>(base.GetInputSlot(0), descriptor, name.c_str()); - // Set tensor info for new layer + // Sets tensor info for new layer. newReshape.GetOutputHandler().SetTensorInfo(outInfo); - // Reconnect base with original parent + // Reconnects base with original parent. newReshape.GetOutputSlot().MoveAllConnections(*parentOut); - // Parent is now the new layer + // Parent is now the new layer. parentOut = &newReshape.GetOutputSlot(); } - // Move connections in child output to parent layer. + // Moves connections in child output to parent layer. // Child layer will be removed as it's left unconnected. // Base layer will be removed if left unconnected. child.GetOutputSlot().MoveAllConnections(*parentOut); |