diff options
author | 2018-08-29 10:34:08 -0700 | |
---|---|---|
committer | 2018-08-29 10:38:34 -0700 | |
commit | 150dee25d82589ca109957cc996efbd2a236e044 (patch) | |
tree | ef608bb703027596ed47ea4f30fa6eab93389005 /tensorflow/compiler/xla/tests/client_library_test_base.cc | |
parent | aca93368a979419360c1fd84b53b1766b19ba81a (diff) |
[XLA] Implement variadic reduce in the evaluator. It is currently supported only for the case where all of the inputs have the same element type.
PiperOrigin-RevId: 210746149
Diffstat (limited to 'tensorflow/compiler/xla/tests/client_library_test_base.cc')
-rw-r--r-- | tensorflow/compiler/xla/tests/client_library_test_base.cc | 6 |
1 files changed, 0 insertions, 6 deletions
diff --git a/tensorflow/compiler/xla/tests/client_library_test_base.cc b/tensorflow/compiler/xla/tests/client_library_test_base.cc index 9cd974fd9b..f1ab83df82 100644 --- a/tensorflow/compiler/xla/tests/client_library_test_base.cc +++ b/tensorflow/compiler/xla/tests/client_library_test_base.cc @@ -290,10 +290,6 @@ Status ClientLibraryTestBase::ComputeAndCompareLiteralWithStatus( if (ShapeUtil::ElementIsFloating(expected.shape()) || ShapeUtil::ElementIsComplex(expected.shape())) { LOG(WARNING) << "performing exact comparison of floating point numbers"; - } else { - TF_RET_CHECK(ShapeUtil::ElementIsIntegral(expected.shape()) || - expected.shape().element_type() == PRED) - << ShapeUtil::HumanString(expected.shape()); } // We allow using a float expected literal for a bfloat16 output. In this // case, we need to convert the expected literal to bfloat16. @@ -350,8 +346,6 @@ Status ClientLibraryTestBase::ComputeAndCompareLiteralWithStatus( } } - TF_RET_CHECK(ShapeUtil::ElementIsFloating(expected.shape()) || - ShapeUtil::ElementIsComplex(expected.shape())); TF_ASSIGN_OR_RETURN(auto computation, builder->Build()); // We allow using a float expected literal for a bfloat16 output. In this // case, we need to convert the expected literal to bfloat16. |