[TF:XLA] Enable tests and update descriptions.

PiperOrigin-RevId: 232637449
This commit is contained in:
A. Unique TensorFlower 2019-02-06 01:53:47 -08:00 committed by TensorFlower Gardener
parent 97c19661e9
commit e5b9fdd4c7

View File

@ -1211,7 +1211,6 @@ class PoolingTest(test.TestCase):
[1, window_rows, window_cols, 1],
[1, row_stride, col_stride, 1], padding)
@test_util.disable_xla("This test never passed for XLA")
def _testMaxPoolGradDirect(self, input_data, output_backprop,
expected_input_backprop, input_sizes, output_sizes,
window_rows, window_cols, row_stride, col_stride,
@ -1353,6 +1352,7 @@ class PoolingTest(test.TestCase):
use_gpu=use_gpu,
v2=v2)
@test_util.disable_xla("b/123923733") # NaNs handled differently
def _testMaxPoolGradDirectWithNans2_1(self):
input_data = [float("nan")] * 16
output_backprop = [11.0, 12.0, 13.0, 15.0, 16.0, 17.0, 19.0, 20.0, 21.0]
@ -1427,6 +1427,7 @@ class PoolingTest(test.TestCase):
else:
del os.environ["TF_ENABLE_MAXPOOL_NANPROP"]
@test_util.disable_xla("b/123923733") # NaNs handled differently
def _testMaxPoolGradDirectWithNans2_2(self):
input_data = [float("nan")] * 16
output_backprop = [
@ -1627,7 +1628,6 @@ class PoolingTest(test.TestCase):
use_gpu=use_gpu)
@test_util.run_deprecated_v1
@test_util.disable_xla("This test never passed for XLA")
def testMaxPoolGradGrad(self):
for (data_format, use_gpu) in GetTestConfigs():
self._testMaxPoolGradGradValidPadding1_1(data_format, use_gpu)
@ -1662,7 +1662,6 @@ class PoolingTest(test.TestCase):
[1, row_stride, col_stride, 1], padding)
@test_util.run_deprecated_v1
@test_util.disable_xla("This test never passed for XLA")
def testAvgPoolGrad(self):
for (data_format, use_gpu) in GetTestConfigs():
self._testAvgPoolGradValidPadding1_1(data_format, use_gpu)
@ -1901,9 +1900,10 @@ if __name__ == "__main__":
if name_ == "maxpool5":
setattr(
PoolingTest, "testMaxPoolGrad_" + name_,
test_util.disable_xla("maxpool5 fails while all others pass")(
GetMaxPoolGradTest(input_size_, filter_size_, output_size_,
stride_, padding_)))
test_util.disable_xla(
"b/123926014: incorrect output with only constants")(
GetMaxPoolGradTest(input_size_, filter_size_, output_size_,
stride_, padding_)))
else:
setattr(
PoolingTest, "testMaxPoolGrad_" + name_,