From d713a4d841ad09a9466e43aa1bf0be09fe54ea22 Mon Sep 17 00:00:00 2001 From: TatWai Chong Date: Thu, 10 Nov 2022 13:54:28 -0800 Subject: Add framework test for math.sign The result comparison between Tensorflow runtime and the reference model hasn't been checked as the sign operator is not supported by the native TFLite runtime. That said, since the generated tosa ops for tf.sign and tfl.sign is identical, the correctness presumably can be proved by the result from tf.sign. Change-Id: I72eb415df7fb6ca4dc9103f9ddc7104b0ba39234 Signed-off-by: TatWai Chong --- verif/frameworks/test_builder.py | 7 +++++++ verif/frameworks/tosa_verif_framework_generator.py | 7 +++++++ 2 files changed, 14 insertions(+) diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py index 6e7b6a5..e86d5fe 100644 --- a/verif/frameworks/test_builder.py +++ b/verif/frameworks/test_builder.py @@ -317,6 +317,13 @@ class TBuilder: def eval(self, a): return tf.math.rsqrt(a, name=self.result_name) + class Sign: + def __init__(self, name): + self.result_name = name + + def eval(self, a): + return tf.math.sign(a, name=self.result_name) + class Sigmoid: def __init__(self, name): self.result_name = name diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 36ddda5..742f991 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -298,6 +298,13 @@ TF_OP_LIST = { "build_fcn": (TBuilder.Rsqrt, TGen.tgBasic, ArgGen.agNone), "types": TYPE_F, }, + "sign": { + "operands": (1, 0), + "build_fcn": (TBuilder.Sign, TGen.tgBasic, ArgGen.agNone), + "types": { + "tf": TYPE_F, + }, + }, "sigmoid": { "operands": (1, 0), "build_fcn": (TBuilder.Sigmoid, TGen.tgBasic, ArgGen.agNone), -- cgit v1.2.1