diff options
Diffstat (limited to 'tensorflow/compiler/xla/tests/client_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/tests/client_test.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/tests/client_test.cc b/tensorflow/compiler/xla/tests/client_test.cc index 0853feeebd..183bcf1dd3 100644 --- a/tensorflow/compiler/xla/tests/client_test.cc +++ b/tensorflow/compiler/xla/tests/client_test.cc @@ -54,8 +54,8 @@ TEST_F(ClientTest, ExecuteWithLayout) { .ConsumeValueOrDie(); std::unique_ptr<Literal> expected_literal = - test_utils::CreateR2LiteralWithLayout<int32>({{11, 22}, {33, 44}}, - transfer_layout); + Literal::CreateR2WithLayout<int32>( + {{11, 22}, {33, 44}}, LayoutUtil::MakeLayout(transfer_layout)); auto computed = client_->Transfer(*data, &expected_literal->shape()); |