message: Making the failure an OpError rather than InternalError because changing the compilation option to --define=with_mlir_support=true changes the place in the placer logic it fails.
diff: @@ -108,9 +108,7 @@ class ReplicateClusterTest(test_base.DatasetTestBase, parameterized.TestCase):
  it1 = dataset_ops.make_initializable_iterator(dataset1)
  # We don't support stateful ops across processes in functions as of now.
  with session.Session(self._target) as sess:
  - with self.assertRaisesRegexp(
  - errors.InternalError,
  - "tf.function with resource inputs residing on remote devices."):
  + with self.assertRaises(errors.OpError):
  sess.run(it1.initializer)
  @combinations.generate(
  