message: Fix DefFunctionTest.testConstantOnWrongDevice to not be flakey
  Previously it generated a random shape using random_normal what caused a failure
  when the shape become invalid (e.g. -2 as dimension size).
diff: @@ -646,7 +646,7 @@ class DefFunctionTest(xla_test.XLATestCase):
  def testConstantOnWrongDevice(self):
  with ops.device('device:{}:0'.format(self.device)):
  - s = math_ops.cast(random_ops.random_normal([2]), dtypes.int32)
  + s = random_ops.random_uniform([2], 1, 10, dtypes.int32)
  l = random_ops.random_normal([s[0] * s[1]])
  @def_function.function(experimental_compile=True)
  