diff options
Diffstat (limited to 'tensorflow/compiler/xla/tests/reduce_precision_test.cc')
| -rw-r--r-- | tensorflow/compiler/xla/tests/reduce_precision_test.cc | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/tensorflow/compiler/xla/tests/reduce_precision_test.cc b/tensorflow/compiler/xla/tests/reduce_precision_test.cc index 4c1aa12106..04c7f31646 100644 --- a/tensorflow/compiler/xla/tests/reduce_precision_test.cc +++ b/tensorflow/compiler/xla/tests/reduce_precision_test.cc @@ -24,7 +24,7 @@ limitations under the License. #include "tensorflow/compiler/xla/client/local_client.h" #include "tensorflow/compiler/xla/client/xla_client/xla_builder.h" #include "tensorflow/compiler/xla/layout_util.h" -#include "tensorflow/compiler/xla/literal_util.h" +#include "tensorflow/compiler/xla/literal.h" #include "tensorflow/compiler/xla/service/reduce_precision_insertion.h" #include "tensorflow/compiler/xla/statusor.h" #include "tensorflow/compiler/xla/test.h" @@ -230,7 +230,8 @@ XLA_TEST_P(ReducePrecisionAccuracyTest, ReducePrecisionF32) { XlaBuilder builder(TestName()); - std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({input_values}); + std::unique_ptr<Literal> a_literal = + LiteralUtil::CreateR1<float>({input_values}); std::unique_ptr<GlobalData> a_data = client_->TransferToServer(*a_literal).ConsumeValueOrDie(); auto a = Parameter(&builder, 0, a_literal->shape(), "a"); @@ -253,7 +254,7 @@ XLA_TEST_F(ReducePrecisionInsertionTest, DISABLED_ON_INTERPRETER(ReducePrecisionBeforeFusion)) { XlaBuilder builder(TestName()); - std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({1.00001}); + std::unique_ptr<Literal> a_literal = LiteralUtil::CreateR1<float>({1.00001}); std::unique_ptr<GlobalData> a_data = client_->TransferToServer(*a_literal).ConsumeValueOrDie(); auto a = Parameter(&builder, 0, a_literal->shape(), "a"); @@ -282,7 +283,7 @@ XLA_TEST_F(ReducePrecisionInsertionTest, DISABLED_ON_INTERPRETER(ReducePrecisionSkippedAfterFusion)) { XlaBuilder builder(TestName()); - std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({1.00001}); + std::unique_ptr<Literal> a_literal = LiteralUtil::CreateR1<float>({1.00001}); std::unique_ptr<GlobalData> a_data = client_->TransferToServer(*a_literal).ConsumeValueOrDie(); auto a = Parameter(&builder, 0, a_literal->shape(), "a"); @@ -308,7 +309,7 @@ XLA_TEST_F(ReducePrecisionInsertionTest, DISABLED_ON_INTERPRETER(ReducePrecisionAddedAfterFusion)) { XlaBuilder builder(TestName()); - std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({1.00001}); + std::unique_ptr<Literal> a_literal = LiteralUtil::CreateR1<float>({1.00001}); std::unique_ptr<GlobalData> a_data = client_->TransferToServer(*a_literal).ConsumeValueOrDie(); auto a = Parameter(&builder, 0, a_literal->shape(), "a"); @@ -332,7 +333,7 @@ XLA_TEST_F(ReducePrecisionInsertionTest, DISABLED_ON_INTERPRETER(ReducePrecisionSkippedFusionContains)) { XlaBuilder builder(TestName()); - std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({1.00001}); + std::unique_ptr<Literal> a_literal = LiteralUtil::CreateR1<float>({1.00001}); std::unique_ptr<GlobalData> a_data = client_->TransferToServer(*a_literal).ConsumeValueOrDie(); auto a = Parameter(&builder, 0, a_literal->shape(), "a"); @@ -357,7 +358,7 @@ XLA_TEST_F(ReducePrecisionInsertionTest, DISABLED_ON_INTERPRETER(ReducePrecisionAddedFusionContains)) { XlaBuilder builder(TestName()); - std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({1.00001}); + std::unique_ptr<Literal> a_literal = LiteralUtil::CreateR1<float>({1.00001}); std::unique_ptr<GlobalData> a_data = client_->TransferToServer(*a_literal).ConsumeValueOrDie(); auto a = Parameter(&builder, 0, a_literal->shape(), "a"); |