aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/optimizations/MovePermuteUp.hpp
diff options
context:
space:
mode:
authortelsoa01 <telmo.soares@arm.com>2018-08-31 09:22:23 +0100
committertelsoa01 <telmo.soares@arm.com>2018-08-31 09:22:23 +0100
commitc577f2c6a3b4ddb6ba87a882723c53a248afbeba (patch)
treebd7d4c148df27f8be6649d313efb24f536b7cf34 /src/armnn/optimizations/MovePermuteUp.hpp
parent4c7098bfeab1ffe1cdc77f6c15548d3e73274746 (diff)
downloadarmnn-c577f2c6a3b4ddb6ba87a882723c53a248afbeba.tar.gz
Release 18.08
Diffstat (limited to 'src/armnn/optimizations/MovePermuteUp.hpp')
-rw-r--r--src/armnn/optimizations/MovePermuteUp.hpp10
1 files changed, 5 insertions, 5 deletions
diff --git a/src/armnn/optimizations/MovePermuteUp.hpp b/src/armnn/optimizations/MovePermuteUp.hpp
index 8c59986762..a8e18f5add 100644
--- a/src/armnn/optimizations/MovePermuteUp.hpp
+++ b/src/armnn/optimizations/MovePermuteUp.hpp
@@ -31,24 +31,24 @@ public:
auto permute = boost::polymorphic_downcast<PermuteLayer*>(&connection.GetOwningLayer());
const PermutationVector& perm = permute->GetPermutation();
- // Insert an equivalent permute before every input of the base layer.
+ // Inserts an equivalent permute before every input of the base layer.
for (auto baseInput = base.BeginInputSlots(); baseInput != base.EndInputSlots(); ++baseInput)
{
- // Insert new permute layer.
+ // Inserts a new permute layer.
const std::string name = std::string("moved_up-") + permute->GetName();
PermuteLayer& permLayer = *graph.InsertNewLayer<PermuteLayer>(*baseInput, perm, name.c_str());
- // Set output tensor info for the new layer.
+ // Sets output tensor info for the new layer.
OutputSlot& parentOutput = *permLayer.GetInputSlot(0).GetConnectedOutputSlot();
const TensorInfo permOutInfo = armnnUtils::Permuted(parentOutput.GetTensorInfo(), perm);
permLayer.GetOutputHandler().SetTensorInfo(permOutInfo);
}
- // Set permuted output tensor info
+ // Sets permuted output tensor info
const TensorInfo& childOutInfo = permute->GetOutputHandler().GetTensorInfo();
base.GetOutputHandler().SetTensorInfo(childOutInfo);
- // Bypass permute. It will be removed as it's left unconnected.
+ // Bypasses permute. It will be removed as it's left unconnected.
permute->GetOutputSlot().MoveAllConnections(base.GetOutputSlot());
}
}