Add tf broadcast-to testing

This patch adds BoradcastTo Op to the tf tests.
Did not add tflite testing because the tf.lite.TFLiteConverter converts
tf.broadcast-to to tfl.mul by 1.

Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: Icd372e619c318121c19eca87d5716bcd5fbbbb23

1 files changed, 8 insertions, 0 deletions

diff --git a/verif/frameworks/test_builder.py b/verif/frameworks/test_builder.py
index f872888..1b681d2 100644
--- a/verif/frameworks/test_builder.py
+++ b/verif/frameworks/test_builder.py
@@ -1239,3 +1239,11 @@ class TBuilder:
 
         def eval(self, a):
             return tf.math.imag(a, name=self.result_name)
+
+    class BroadcastTo:
+        def __init__(self, shape, name):
+            self.shape = shape
+            self.result_name = name
+
+        def eval(self, a):
+            return tf.broadcast_to(a, shape=self.shape, name=self.result_name)