Function XLA_TEST_F

tensorflow/compiler/xla/tests/transfer_manager_test.cc:71–84

};

XLA_TEST_F(TransferManagerTest, TransferR0U32) {
  Literal literal = LiteralUtil::CreateR0<uint32>(42);
  const Shape& shape = literal.shape();
  auto device_buffer = AllocateDeviceBuffer(shape);

  // Round trip literal through device.
  ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal,
                                                          device_buffer));
  TF_ASSERT_OK_AND_ASSIGN(
      Literal result,
      transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer));

  LiteralTestUtil::ExpectR0Equal<uint32>(42, result);
}

XLA_TEST_F(TransferManagerTest, TransferR1F32) {
  Literal literal =