OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 12:20:24 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Make Service::TransferToClient not attempt to manipulate the literal when the transfer failed, preventing a crash and allowing the caller to see the reason for the failed transfer.

Browse Source 
PiperOrigin-RevId: 169770126
This commit is contained in:
Bjarke Hammersholt Roune 2017-09-22 20:38:54 -07:00 committed by TensorFlower Gardener
parent 8ef722253c
commit 286f57061c
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/compiler/xla/service/service.cc
View File

@ -930,9 +930,10 @@ tensorflow::Status Service::TransferToClient(const TransferToClientRequest* arg,
} }
Literal literal; Literal literal;
auto status = LiteralFromAllocation(allocation, *literal_shape, &literal); TF_RETURN_IF_ERROR(
LiteralFromAllocation(allocation, *literal_shape, &literal));
*result->mutable_literal() = literal.ToProto(); *result->mutable_literal() = literal.ToProto();
return status; return tensorflow::Status::OK();
} }
tensorflow::Status Service::TransferToServer(const TransferToServerRequest* arg, tensorflow::Status Service::TransferToServer(const TransferToServerRequest* arg,