message: Add test to ensure stateful experimental_run_v2 is always executed.
diff: @@ -96,6 +96,28 @@ class InputIterationTest(test.TestCase, parameterized.TestCase,
  constant_op.constant(4., shape=(distribution.num_replicas_in_sync)),
  run(2.))
  + @combinations.generate(
  + combinations.combine(
  + distribution=strategy_combinations.all_strategies,
  + mode=["eager"]
  + ))
  + def testStatefulExperimentalRunAlwaysExecute(self, distribution):
  + with distribution.scope():
  + v = variables.Variable(
  + 0.0, aggregation=variables.VariableAggregation.MEAN)
  +
  + @def_function.function
  + def train_step():
  +
  + def assign_add():
  + v.assign_add(1.0)
  +
  + distribution.experimental_run_v2(assign_add)
  + return array_ops.zeros([])
  +
  + train_step()
  + self.assertAllEqual(1.0, v.numpy())
  +
  @combinations.generate(
  combinations.combine(
  distribution=strategy_combinations.strategies_minus_tpu,
  