diff options
Diffstat (limited to 'src/armnn/optimizations/AddBroadcastReshapeLayer.hpp')
-rw-r--r-- | src/armnn/optimizations/AddBroadcastReshapeLayer.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp b/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp index 4cfe2e4898..d243a807fd 100644 --- a/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp +++ b/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp @@ -17,7 +17,8 @@ namespace optimizations static const std::set<armnn::LayerType> broadcastOps{ LayerType::Addition, LayerType::Division, LayerType::Maximum, LayerType::Minimum, - LayerType::Multiplication, LayerType::Subtraction }; + LayerType::Multiplication, LayerType::Prelu, + LayerType::Subtraction }; class AddBroadcastReshapeLayerImpl { |