From f7008da16ed36fce2866e0a4a2595acc8f0a27d6 Mon Sep 17 00:00:00 2001 From: TatWai Chong Date: Fri, 9 Sep 2022 09:35:40 +0000 Subject: Add framework test for TF and TFL mirrorpad Signed-off-by: TatWai Chong Change-Id: Icc9b8f6a65ee54ddbb445c3a999ca49401b660c2 --- verif/frameworks/test_builder.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'verif/frameworks/test_builder.py') 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 -- cgit v1.2.1