diff options
author | Cathal Corbett <cathal.corbett@arm.com> | 2022-05-16 15:20:56 +0100 |
---|---|---|
committer | Ryan OShea <ryan.oshea3@arm.com> | 2022-05-19 11:07:09 +0100 |
commit | 630ce65543c08d8e7fca5be80f9a64122744d135 (patch) | |
tree | f78db6d55bc9feaf884d83212c9ec9195f482225 /src/armnn/optimizations/All.hpp | |
parent | 5841c740ba6bc6c8c3e96d24156dc47907af6430 (diff) | |
download | armnn-630ce65543c08d8e7fca5be80f9a64122744d135.tar.gz |
IVGCVSW-6147 ConstTensorsAsInput: Optimizer - FusePermuteIntoConstLayer
* No trailing permute layer after a constant layer
* Unit test for optimization
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: I0d098f5af41d2c55df7cef1ccfb848093320ddc1
Diffstat (limited to 'src/armnn/optimizations/All.hpp')
-rw-r--r-- | src/armnn/optimizations/All.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnn/optimizations/All.hpp b/src/armnn/optimizations/All.hpp index e4a1f33e08..900e763762 100644 --- a/src/armnn/optimizations/All.hpp +++ b/src/armnn/optimizations/All.hpp @@ -8,6 +8,7 @@ #include "AddDebug.hpp" #include "ConvertConstants.hpp" #include "ConvertConstDequantisationLayersToConstLayers.hpp" +#include "ConvertConstPermuteLayersToConstLayers.hpp" #include "ConvertFp32NetworkToBf16.hpp" #include "ConvertFp32NetworkToFp16.hpp" #include "FoldPadIntoLayer2d.hpp" |