diff options
| author |  Anna R <annarev@google.com> | 2018-09-25 11:50:05 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-25 11:58:31 -0700 |
| commit | 3f2d0ad596d5f2cc30f10148ac56ac5680603f5b (patch) | |
| tree | 97fdc98c0378a13b750712a5af8de6cc4539147c /tensorflow/python/kernel_tests | |
| parent | dfafefa3054a37b64d0d47419eb3f7a576e662db (diff) | |
Temporarily disable float16 tests in depthwise_conv_op_test. They seem to be failing when running with P100.
PiperOrigin-RevId: 214477405
Diffstat (limited to 'tensorflow/python/kernel_tests')
| -rw-r--r-- | tensorflow/python/kernel_tests/depthwise_conv_op_test.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/python/kernel_tests/depthwise_conv_op_test.py b/tensorflow/python/kernel_tests/depthwise_conv_op_test.py index 200da772e5..6d1ead20be 100644 --- a/tensorflow/python/kernel_tests/depthwise_conv_op_test.py +++ b/tensorflow/python/kernel_tests/depthwise_conv_op_test.py @@ -191,7 +191,7 @@ class DepthwiseConv2DTest(test.TestCase): tf_logging.info( "Testing DepthwiseConv2D, %dth config: %r * %r, stride: %d, padding: " "%s", index, input_size, filter_size, stride, padding) - for data_type in [dtypes.float16, dtypes.float32, dtypes.float64]: + for data_type in [dtypes.float32, dtypes.float64]: tf_logging.info("Testing without grouped_conv") self._VerifyValues( input_size, filter_size, stride, padding, data_type, use_gpu=True) @@ -227,7 +227,7 @@ class DepthwiseConv2DTest(test.TestCase): tf_logging.info( "Testing DepthwiseConv2DFormat, %dth config: %r * %r, stride: %d, " "padding: %s", index, input_size, filter_size, stride, padding) - for data_type in [dtypes.float16, dtypes.float32, dtypes.float64]: + for data_type in [dtypes.float32, dtypes.float64]: self._VerifyValues( input_size, filter_size, @@ -434,7 +434,7 @@ class DepthwiseConv2DTest(test.TestCase): tf_logging.info( "Testing DepthwiseConv2DInputGrad, %dth config: %r * %r, stride: %d, " "padding: %s", index, input_size, filter_size, stride, padding) - for data_type in [dtypes.float16, dtypes.float32, dtypes.float64]: + for data_type in [dtypes.float32, dtypes.float64]: self._ConstructAndTestGradient( input_size, filter_size, @@ -465,7 +465,7 @@ class DepthwiseConv2DTest(test.TestCase): "Testing DepthwiseConv2DInputGradFormat, %dth config: %r * %r, " "stride: %d, padding: %s", index, input_size, filter_size, stride, padding) - for data_type in [dtypes.float16, dtypes.float32, dtypes.float64]: + for data_type in [dtypes.float32, dtypes.float64]: self._ConstructAndTestGradient( input_size, filter_size, @@ -483,7 +483,7 @@ class DepthwiseConv2DTest(test.TestCase): tf_logging.info( "Testing DepthwiseConv2DFilterGrad, %dth config: %r * %r, stride: " "%d, padding: %s", index, input_size, filter_size, stride, padding) - for data_type in [dtypes.float16, dtypes.float32, dtypes.float64]: + for data_type in [dtypes.float32, dtypes.float64]: self._ConstructAndTestGradient( input_size, filter_size, @@ -504,7 +504,7 @@ class DepthwiseConv2DTest(test.TestCase): "Testing DepthwiseConv2DFilterGradFormat, %dth config: %r * %r, " "stride: %d, padding: %s", index, input_size, filter_size, stride, padding) - for data_type in [dtypes.float16, dtypes.float32, dtypes.float64]: + for data_type in [dtypes.float32, dtypes.float64]: self._ConstructAndTestGradient( input_size, filter_size, |