diff options
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 |