[tf.data] Unskip placement test with explicit placement
PiperOrigin-RevId: 333365327 Change-Id: Ic3760551eb5ed5053b4b46513c8882ebc9e72a8b
This commit is contained in:
parent
c4e48908f9
commit
705dcc359c
@ -76,7 +76,6 @@ class PlacementTest(test_base.DatasetTestBase, parameterized.TestCase):
|
|||||||
self.assertEqual(f().numpy(), 45)
|
self.assertEqual(f().numpy(), 45)
|
||||||
|
|
||||||
def testCondWithPlacement(self):
|
def testCondWithPlacement(self):
|
||||||
self.skipTest("b/166625126")
|
|
||||||
# When the cond op is explicitly placed, there shouldn't be cross-device
|
# When the cond op is explicitly placed, there shouldn't be cross-device
|
||||||
# copies.
|
# copies.
|
||||||
@def_function.function
|
@def_function.function
|
||||||
|
Loading…
x
Reference in New Issue
Block a user