message: Fix X64 outputs with dynamic shapes.
diff: @@ -533,6 +533,31 @@ class InputIterationTest(test.TestCase, parameterized.TestCase,
  # This assumes that there are exactly 2 replicas
  self.assertAllEqual([5.5, 7.], run(input_iterator))
  + @combinations.generate(
  + combinations.combine(
  + distribution=strategy_combinations.multidevice_strategies,
  + mode=["eager"]))
  + def testDynamicOutputsWithX64(self, distribution):
  + dataset = get_dataset_from_tensor_slices(
  + [5]).map(lambda x: math_ops.cast(x, dtypes.int64)).batch(2)
  + input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
  +
  + @def_function.function
  + def run(iterator):
  +
  + def computation(x):
  + return math_ops.add(x, x)
  +
  + inputs = next(iterator)
  + outputs = distribution.experimental_local_results(
  + distribution.run(computation, args=(inputs,)))
  + return outputs
  +
  + # This assumes that there are exactly 2 replicas
  + result = run(input_iterator)
  + self.assertAllEqual([10], result[0])
  + self.assertAllEqual([], result[1])
  +
  @combinations.generate(
  combinations.combine(
  distribution=strategy_combinations.multidevice_strategies,
  