message: Re-enable nested function test on TPU. Underlying issue was resolved previously.
diff: @@ -23,7 +23,6 @@ from tensorflow.python import tf2
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.distribute import combinations
  from tensorflow.python.distribute import strategy_combinations
  -from tensorflow.python.distribute import tpu_strategy
  from tensorflow.python.eager import backprop
  from tensorflow.python.eager import def_function
  from tensorflow.python.eager import test
  @@ -246,12 +245,7 @@ class GradientTapeTest(test.TestCase, parameterized.TestCase):
  mode=["eager"],
  model_in_tf_function=[True, False]
  ))
  - def test1(self, distribution, model_in_tf_function):
  - # b/134975331
  - if model_in_tf_function and isinstance(
  - distribution, (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
  - self.skipTest("model inside tf.function doesn't work with TPUStrategy")
  -
  + def testNestedFunction(self, distribution, model_in_tf_function):
  def model(x):
  return x * x
  