2 files changed, 16 insertions, 0 deletions
diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py
index 1b681d2..fcd72a3 100644
--- a/verif/frameworks/test_builder.py
+++ b/verif/frameworks/test_builder.py
@@ -348,6 +348,15 @@ class TBuilder:
         def eval(self, a):
             return tf.math.tanh(a, name=self.result_name)
 
+    class Erf:
+        # tfl.ops cannot be generated right now.
+        # https://github.com/tensorflow/tensorflow/issues/60809
+        def __init__(self, name):
+            self.result_name = name
+
+        def eval(self, a):
+            return tf.math.erf(a, name=self.result_name)
+
     class Sin:
         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 ccbe742..02ab8aa 100755
--- a/verif/frameworks/tosa_verif_framework_generator.py
+++ b/verif/frameworks/tosa_verif_framework_generator.py
@@ -338,6 +338,13 @@ TF_OP_LIST = {
             ),
         },
     },
+    "erf": {
+        "operands": (1, 0),
+        "build_fcn": (TBuilder.Erf, TGen.tgBasic, ArgGen.agNone),
+        "types": {
+            "tf": TYPE_F,
+        },
+    },
     "sin": {
         "operands": (1, 0),
         "build_fcn": (TBuilder.Sin, TGen.tgBasic, ArgGen.agNone),