1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc/grpc_worker_service.cc b/tensorflow/core/distributed_runtime/rpc/grpc_worker_service.cc
index 3bee20623b..13aeb9f9c7 100644
--- a/tensorflow/core/distributed_runtime/rpc/grpc_worker_service.cc
+++ b/tensorflow/core/distributed_runtime/rpc/grpc_worker_service.cc
@@ -377,7 +377,7 @@ class GrpcWorkerService : public AsyncServiceInterface {
               recv->set_key(key);
               // TODO(zhifengc): Deal with gpu -> cpu copy.
               TensorProto* proto = recv->mutable_val();
-              val.AsProtoField(proto);
+              val.AsProtoTensorContent(proto);
             }
           }
           delete collector;