Add framework test for TF and TFL mirrorpad

Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
Change-Id: Icc9b8f6a65ee54ddbb445c3a999ca49401b660c2

1 files changed, 15 insertions, 0 deletions

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 @@ class TBuilder:
                 )
             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