Add framework test for TF and TFL mirrorpad
Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
Change-Id: Icc9b8f6a65ee54ddbb445c3a999ca49401b660c2
diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py
index 0468518..cd7831d 100644
--- a/verif/frameworks/test_builder.py
+++ b/verif/frameworks/test_builder.py
@@ -737,6 +737,21 @@
)
return tf.stack(sums, 0, name=self.result_name)
+ class MirrorPad:
+ def __init__(self, padding, mode, name):
+ self.padding = padding
+ self.mode = mode
+ self.result_name = name
+
+ def eval(self, a):
+ return tf.pad(
+ a,
+ self.padding,
+ mode=self.mode,
+ constant_values=0,
+ name=self.result_name,
+ )
+
class Pad:
def __init__(self, padding, name):
self.padding = padding