message: [XLA] Propogate errors in ClientLibraryTestBase::ComputeAnd*
  Errors returned by Client::TransferToServer should propagate instead of
  triggering a CHECK-failure.
diff: @@ -15,6 +15,7 @@ limitations under the License.
  #include "tensorflow/compiler/xla/tests/client_library_test_base.h"
  +#include <memory>
  #include <string>
  #include "absl/memory/memory.h"
  @@ -279,9 +280,10 @@ StatusOr<Literal> ClientLibraryTestBase::ComputeAndTransfer(
  if (!arguments_.empty()) {
  CHECK(arguments.empty());
  for (const auto& argument : arguments_) {
  - owning_arguments.push_back(
  - client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument))
  - .ValueOrDie());
  + TF_ASSIGN_OR_RETURN(
  + std::unique_ptr<GlobalData> owned_argument,
  + client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument)));
  + owning_arguments.push_back(std::move(owned_argument));
  arguments.push_back(owning_arguments.back().get());
  }
  }
  @@ -302,9 +304,10 @@ Status ClientLibraryTestBase::ComputeAndCompareLiteralWithStatus(
  if (!arguments_.empty()) {
  CHECK(arguments.empty());
  for (const auto& argument : arguments_) {
  - owning_arguments.push_back(
  - client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument))
  - .ValueOrDie());
  + TF_ASSIGN_OR_RETURN(
  + std::unique_ptr<GlobalData> owned_argument,
  + client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument)));
  + owning_arguments.push_back(std::move(owned_argument));
  arguments.push_back(owning_arguments.back().get());
  }
  }
  @@ -362,9 +365,10 @@ Status ClientLibraryTestBase::ComputeAndCompareLiteralWithStatus(
  if (!arguments_.empty()) {
  CHECK(arguments.empty());
  for (const auto& argument : arguments_) {
  - owning_arguments.push_back(
  - client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument))
  - .ValueOrDie());
  + TF_ASSIGN_OR_RETURN(
  + std::unique_ptr<GlobalData> owned_argument,
  + client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument)));
  + owning_arguments.push_back(std::move(owned_argument));
  arguments.push_back(owning_arguments.back().get());
  }
  }
  