diff options
Diffstat (limited to 'verif/frameworks/test_builder.py')
-rw-r--r-- | verif/frameworks/test_builder.py | 15 |
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 |