Whitelist XlaBroadcastHelperOp and enable tests XlaDynamicUpdateSliceOp

PiperOrigin-RevId: 311049106
Change-Id: I6f47f784f744ba3e60f3f377fa90412b1114d3b5

tensorflow/compiler/mlir/xla/transforms/legalize_tf_with_tf2xla.cc

@@ -165,6 +165,7 @@ static bool IsOpWhitelisted(Operation* op) {
     TypeID::get<TF::TruncateModOp>(),
     TypeID::get<TF::UnpackOp>(),
     TypeID::get<TF::XdivyOp>(),
+    TypeID::get<TF::XlaBroadcastHelperOp>(),
     TypeID::get<TF::XlaConvOp>(),
     TypeID::get<TF::XlaDotOp>(),
     TypeID::get<TF::XlaPadOp>(),

tensorflow/compiler/tests/BUILD

@@ -562,6 +562,7 @@ tf_xla_py_test(
     name = "dynamic_slice_ops_test",
     size = "small",
     srcs = ["dynamic_slice_ops_test.py"],
+    enable_mlir_bridge = True,
     python_version = "PY3",
     tags = [
         "no_pip",  # TODO(b/149738646): fix pip install so these tests run on kokoro pip

tensorflow/compiler/tests/xla_ops_test.py

@@ -51,7 +51,6 @@ class XlaOpsNumericalTest(xla_test.XLATestCase, parameterized.TestCase):
         equality_fn = self.assertAllClose
       equality_fn(result, expected, rtol=1e-3)
 
-  @test_util.disable_mlir_bridge('Not supported yet')
   def testAdd(self):
     for dtype in self.numeric_types:
       self._assertOpOutputMatchesExpected(