diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-05-25 14:26:24 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-05-25 15:05:06 +0100 |
commit | 4a4af11f2e02465df8f5aa2b2de19d00c2a8ea3d (patch) | |
tree | d9205277fff0528310f36859dbd76aea7e7f17f3 /src/armnn/optimizations | |
parent | bfaee6b574301a54eab07a6021c39ae710977f7f (diff) | |
download | armnn-4a4af11f2e02465df8f5aa2b2de19d00c2a8ea3d.tar.gz |
IVGCVSW-3649 Add Prelu with different alpha dimension test to TfLiteParser
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I982ecd66ea3ed4d88934cd8254832eecb4a7adb4
Diffstat (limited to 'src/armnn/optimizations')
-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 { |