experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Whitelist XlaBroadcastHelperOp and enable tests XlaDynamicUpdateSliceOp 

PiperOrigin-RevId: 311049106
Change-Id: I6f47f784f744ba3e60f3f377fa90412b1114d3b5
This commit is contained in:
Smit Hinsu 2020-05-11 20:24:30 -07:00 committed by TensorFlower Gardener
parent 875f71cf8b
commit 6598d11e8b
3 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/mlir/xla/transforms/legalize_tf_with_tf2xla.cc
View File

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

1
tensorflow/compiler/tests/BUILD
View File

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

1
tensorflow/compiler/tests/xla_ops_test.py
View File

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