message: Skip conv3d tests of auto-mixed precision for now
diff: @@ -410,11 +410,13 @@ class AutoMixedPrecisionTest(test.TestCase):
  self.assertEqual(num_to_fp32, 1) # After FusedBatchNormV3:0
  self.assertAllClose(output_val_ref, output_val, atol=1e-3, rtol=1e-3)
  + # TODO: enable these tests when cuDNN is upgraded to >= 7.6.2. Same with the
  + # test_conv3d() below.
  + @unittest.skip("Test case should be skipped when cuDNN < 7.6.2")
  @test_util.run_deprecated_v1
  @test_util.disable_xla('This test does not pass with XLA')
  def test_conv3d_bn(self):
  """Test graph with convolution followed by batch norm."""
  - with compat.forward_compatibility_horizon(2019, 6, 7):
  if test.is_gpu_available(cuda_only=True):
  random_seed.set_random_seed(0)
  x = _input([2, 8, 8, 8, 1])
  @@ -433,6 +435,7 @@ class AutoMixedPrecisionTest(test.TestCase):
  self.assertEqual(num_to_fp32, 1) # After FusedBatchNormV3:0
  self.assertAllClose(output_val_ref, output_val, atol=1e-2, rtol=1e-2)
  + @unittest.skip("Test case should be skipped when cuDNN < 7.6.2")
  @test_util.run_deprecated_v1
  @test_util.disable_xla('This test does not pass with XLA')
  def test_conv3d(self):
  