From 4a4af11f2e02465df8f5aa2b2de19d00c2a8ea3d Mon Sep 17 00:00:00 2001
From: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Date: Tue, 25 May 2021 14:26:24 +0100
Subject: IVGCVSW-3649 Add Prelu with different alpha dimension test to
 TfLiteParser

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I982ecd66ea3ed4d88934cd8254832eecb4a7adb4
---
 src/armnn/optimizations/AddBroadcastReshapeLayer.hpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src/armnn/optimizations/AddBroadcastReshapeLayer.hpp')

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
 {
-- 
cgit v1.2.1