aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
-rw-r--r--tensorflow/compiler/tests/adam_test.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/adam_test.py b/tensorflow/compiler/tests/adam_test.py
index ee56a38f94..b904e6676b 100644
--- a/tensorflow/compiler/tests/adam_test.py
+++ b/tensorflow/compiler/tests/adam_test.py
@@ -52,6 +52,7 @@ class AdamOptimizerTest(XLATestCase):
def testBasic(self):
for dtype in self.float_types:
+ # TODO: test fails for float16 due to excessive precision requirements.
if dtype == np.float16:
continue
with self.test_session(), self.test_scope():
@@ -93,6 +94,7 @@ class AdamOptimizerTest(XLATestCase):
def testTensorLearningRate(self):
for dtype in self.float_types:
+ # TODO: test fails for float16 due to excessive precision requirements.
if dtype == np.float16:
continue
with self.test_session(), self.test_scope():
@@ -134,6 +136,7 @@ class AdamOptimizerTest(XLATestCase):
def testSharing(self):
for dtype in self.float_types:
+ # TODO: test fails for float16 due to excessive precision requirements.
if dtype == np.float16:
continue
with self.test_session(), self.test_scope():