From 2eea5bf8b213ce5f7d335cc45686a80cac119cc2 Mon Sep 17 00:00:00 2001 From: Jerry Ge Date: Tue, 11 Oct 2022 16:27:05 +0000 Subject: Add framework tests for TFL.Relu0To1Op Signed-off-by: Jerry Ge Change-Id: I6fa14e9dbe7c0cd21156a5779d68fd473af322f2 --- verif/frameworks/test_builder.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'verif/frameworks/test_builder.py') diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py index c7ba9a9..6302865 100644 --- a/verif/frameworks/test_builder.py +++ b/verif/frameworks/test_builder.py @@ -86,6 +86,16 @@ class TBuilder: # alternatively, we can use clip_by_value return tf.math.minimum(1.0, tf.math.maximum(-1.0, a)) + class Relu0To1: + def __init__(self, name): + self.result_name = name + + def eval(self, a): + # TF doesn't have relu_0_to_1 operator, + # use min and max as a workaround + # alternatively, we can use clip_by_value + return tf.math.minimum(1.0, tf.math.maximum(0.0, a)) + class Relu6: def __init__(self, name): self.result_name = name -- cgit v1.2.1