diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2024-02-21 14:07:41 +0000 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2024-03-08 15:43:37 +0000 |
| commit | 5cc8e447a63d9eb32f0e5ab6c66e755a52fc584d (patch) | |
| tree | 29fa33969a978409c8c8d51bad86123a52cb5513 /src/armnn/Network.cpp | |
| parent | e4ccada2ed349f22cf75c4b8d8d6ba9dc3990694 (diff) | |
| download | armnn-5cc8e447a63d9eb32f0e5ab6c66e755a52fc584d.tar.gz | |
IVGCVSW-8299 Optimisation to replace Max + Min with Bounded Relu
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I0774a9580710350bd0a959cf68dfe057f52e4330
Diffstat (limited to 'src/armnn/Network.cpp')
| -rw-r--r-- | src/armnn/Network.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 60df27d7fc..1a4fec59ce 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -1968,8 +1968,12 @@ IOptimizedNetworkPtr Optimize(const Graph& inGraph, optGraph.InferTensorInfos(); } - // Perform BroadcastToOptimizationLayer and then AddBroadcastReshapeLayer optimisation using namespace optimizations; + // Substitute Max + Min with Bounded Relu before AddBroadcastReshapeLayer optimisation, + // as Bounded ReLu needs the constants to be 1D size 1 + Optimizer::Pass(optGraph, MakeOptimizations(MaxMinIntoBoundedRelu())); + + // Perform BroadcastToOptimizationLayer before AddBroadcastReshapeLayer optimisation Optimizer::Pass(optGraph, MakeOptimizations(BroadcastToOptimizationLayer())); Optimizer::Pass(optGraph, MakeOptimizations(AddBroadcastReshapeLayer())); @@ -1980,7 +1984,6 @@ IOptimizedNetworkPtr Optimize(const Graph& inGraph, optGraph.InferTensorInfos(); } - // Group Constant Layer optimizations together where possible. // This is important as: // FusePermuteIntoConstantLayer must happen before FoldPadIntoDepthwiseConvolution2d and |