experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

re-enabling a subtest withing image_ops_test.py, because it no longer fails on the ROCm platform

Browse Source
This commit is contained in:
Deven Desai 2021-01-07 15:07:36 +00:00
parent d6b25dcbe0
commit 98aecfccda
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/tests/image_ops_test.py
View File

@ -605,9 +605,6 @@ class ResizeBilinearTest(parameterized.TestCase, xla_test.XLATestCase):
)
def test(self, src_y, src_x, dst_y, dst_x, dtype=np.float32):
if test.is_built_with_rocm():
self.skipTest("Disabled on ROCm, because it runs out of memory")
max_y = max(src_y - 1, 1) * (dst_y - 1) + 1
max_x = max(src_x - 1, 1) * (dst_x - 1) + 1