[XLA] Fix use-of-unintialized-value msan failure in local_client as well.

PiperOrigin-RevId: 208004791
This commit is contained in:
Kay Zhu 2018-08-09 00:20:23 -07:00 committed by TensorFlower Gardener
parent f2494ede82
commit b67b3927a1

View File

@ -303,7 +303,7 @@ StatusOr<std::unique_ptr<Literal>> LocalClient::TransferFromOutfeedLocal(
const Shape& shape, int device_ordinal) {
TF_ASSIGN_OR_RETURN(se::StreamExecutor * executor,
backend().stream_executor(device_ordinal));
auto literal = MakeUnique<Literal>(shape);
auto literal = Literal::CreateFromShape(shape);
TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralFromOutfeed(
executor, shape, literal.get()));
return std::move(literal);