[TF:XLA] Enable tests and update descriptions.
PiperOrigin-RevId: 232637449
This commit is contained in:
parent
97c19661e9
commit
e5b9fdd4c7
@ -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_,
|
||||
|
Loading…
x
Reference in New Issue
Block a user