Function TEST

tensorflow/compiler/xla/python/shared_device_buffer_test.cc:27–40

namespace {

TEST(SharedDeviceBufferTest, MakeArray) {
  LocalClient* client = ClientLibrary::LocalClientOrDie();

  Shape shape = ShapeUtil::MakeShape(F32, {3, 101, 4});
  TF_ASSERT_OK_AND_ASSIGN(
      auto buffer, SharedDeviceBuffer::MakeArray(
                       shape, client->backend().transfer_manager(),
                       client->backend().memory_allocator(), 0, nullptr));
  EXPECT_EQ(buffer->children().size(), 0);
  EXPECT_EQ(buffer->device_ordinal(), 0);
  EXPECT_EQ(buffer->allocator(), client->backend().memory_allocator());
  ASSERT_EQ(buffer->device_memory().size(), 1);
  EXPECT_FALSE(buffer->device_memory()[0].is_null());
}

TEST(SharedDeviceBufferTest, MakeTuple) {
  LocalClient* client = ClientLibrary::LocalClientOrDie();