message: patching PR 18797
  Swtich to use axis instead of squeeze_dims in tf.squeeze
diff: @@ -128,18 +128,18 @@ class TestUpgrade(test_util.TensorFlowTestCase):
  self.assertAllEqual(
  tf.expand_dims(
  tf.squeeze(
  - [[1, 2, 3]], squeeze_dims=[0]), dim=0).eval(),
  + [[1, 2, 3]], axis=[0]), dim=0).eval(),
  a)
  self.assertAllEqual(
  tf.squeeze(
  tf.expand_dims(
  - [[1, 2, 3]], dim=1), squeeze_dims=[1]).eval(),
  + [[1, 2, 3]], dim=1), axis=[1]).eval(),
  a)
  self.assertAllEqual(
  tf.squeeze(
  tf.expand_dims(
  - [[1, 2, 3]], dim=1), squeeze_dims=[1]).eval(),
  + [[1, 2, 3]], dim=1), axis=[1]).eval(),
  a)
  @test_util.run_v1_only("b/120545219")
  