From c577f2c6a3b4ddb6ba87a882723c53a248afbeba Mon Sep 17 00:00:00 2001 From: telsoa01 Date: Fri, 31 Aug 2018 09:22:23 +0100 Subject: Release 18.08 --- src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp') 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(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); -- cgit v1.2.1