diff options
author | 2017-09-12 20:10:34 -0700 | |
---|---|---|
committer | 2017-09-12 20:14:31 -0700 | |
commit | 11d3ac29d526a15a2762e864e98ea42c87a34478 (patch) | |
tree | a63eec8617cde989e10e4d40b36590ab5f3a70f6 /tensorflow/compiler/xla/tests/client_library_test_base.cc | |
parent | 3cd6bdef5fa44efbf2b16eeb5fe026be839e6898 (diff) |
[XLA] Add tests for large numbers of parameter / return values and while loops.
PiperOrigin-RevId: 168487225
Diffstat (limited to 'tensorflow/compiler/xla/tests/client_library_test_base.cc')
-rw-r--r-- | tensorflow/compiler/xla/tests/client_library_test_base.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/tests/client_library_test_base.cc b/tensorflow/compiler/xla/tests/client_library_test_base.cc index 3001813dd4..9f3b66e256 100644 --- a/tensorflow/compiler/xla/tests/client_library_test_base.cc +++ b/tensorflow/compiler/xla/tests/client_library_test_base.cc @@ -258,7 +258,8 @@ tensorflow::Status ClientLibraryTestBase::ComputeAndCompareLiteralWithStatus( LOG(WARNING) << "performing exact comparison of floating point numbers"; } else { TF_RET_CHECK(ShapeUtil::ElementIsIntegral(expected.shape()) || - expected.shape().element_type() == PRED); + expected.shape().element_type() == PRED) + << ShapeUtil::HumanString(expected.shape()); } auto expect_equal = [&](const Literal& actual, const string& error_message) { LiteralTestUtil::ExpectEqual(expected, actual, error_message); |