Add framework test for tfl.prelu
Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
Change-Id: Ib894e50f4b869a3f4936885704995a417ea25eef
diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py
index cd7831d..b282287 100644
--- a/verif/frameworks/test_builder.py
+++ b/verif/frameworks/test_builder.py
@@ -101,6 +101,18 @@
def eval(self, a):
return tf.nn.leaky_relu(a, alpha=self.alpha, name=self.result_name)
+ class Prelu:
+ def __init__(self, name):
+ self.result_name = name
+ self.prelu = tf.keras.layers.PReLU(
+ alpha_initializer=tf.keras.initializers.RandomNormal(
+ mean=0.0, stddev=1.0
+ )
+ )
+
+ def eval(self, a):
+ return self.prelu(a)
+
class Gelu:
def __init__(self, name):
self.result_name = name