diff options
Diffstat (limited to 'src/armnn/optimizations/SquashEqualSiblings.hpp')
| -rw-r--r-- | src/armnn/optimizations/SquashEqualSiblings.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/optimizations/SquashEqualSiblings.hpp b/src/armnn/optimizations/SquashEqualSiblings.hpp index 12637ba9b6..d5a8a5d81e 100644 --- a/src/armnn/optimizations/SquashEqualSiblings.hpp +++ b/src/armnn/optimizations/SquashEqualSiblings.hpp @@ -64,6 +64,8 @@ protected: }; using SquashEqualPermuteSiblings = OptimizeForConnection<Layer, PermuteLayer, SquashEqualSiblingsImpl<PermuteLayer>>; +using SquashEqualTransposeSiblings = OptimizeForConnection<Layer, TransposeLayer, + SquashEqualSiblingsImpl<TransposeLayer>>; using SquashEqualReshapeSiblings = OptimizeForConnection<Layer, ReshapeLayer, SquashEqualSiblingsImpl<ReshapeLayer>>; } // namespace optimizations |