diff options
Diffstat (limited to 'tensorflow/core/distributed_runtime/rpc/grpc_session.cc')
-rw-r--r-- | tensorflow/core/distributed_runtime/rpc/grpc_session.cc | 15 |
1 files changed, 6 insertions, 9 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc/grpc_session.cc b/tensorflow/core/distributed_runtime/rpc/grpc_session.cc index fd1c150fa7..fdce1b10e0 100644 --- a/tensorflow/core/distributed_runtime/rpc/grpc_session.cc +++ b/tensorflow/core/distributed_runtime/rpc/grpc_session.cc @@ -452,15 +452,12 @@ class GrpcSessionFactory : public SessionFactory { return str_util::StartsWith(options.target, kSchemePrefix); } - Session* NewSession(const SessionOptions& options) override { - std::unique_ptr<GrpcSession> ret; - Status s = GrpcSession::Create(options, &ret); - if (s.ok()) { - return ret.release(); - } else { - LOG(ERROR) << "Error during session construction: " << s.ToString(); - return nullptr; - } + Status NewSession(const SessionOptions& options, + Session** out_session) override { + std::unique_ptr<GrpcSession> session; + TF_RETURN_IF_ERROR(GrpcSession::Create(options, &session)); + *out_session = session.release(); + return Status::OK(); } // Invokes the session specific static method to reset containers. |