diff options
Diffstat (limited to 'tensorflow/core/common_runtime/session.cc')
-rw-r--r-- | tensorflow/core/common_runtime/session.cc | 20 |
1 files changed, 13 insertions, 7 deletions
diff --git a/tensorflow/core/common_runtime/session.cc b/tensorflow/core/common_runtime/session.cc index 4a9248171b..8c30beeec2 100644 --- a/tensorflow/core/common_runtime/session.cc +++ b/tensorflow/core/common_runtime/session.cc @@ -53,27 +53,33 @@ Status Session::PRun(const string& handle, Session* NewSession(const SessionOptions& options) { SessionFactory* factory; - const Status s = SessionFactory::GetFactory(options, &factory); + Status s = SessionFactory::GetFactory(options, &factory); if (!s.ok()) { LOG(ERROR) << s; return nullptr; } - return factory->NewSession(options); + Session* out_session; + s = NewSession(options, &out_session); + if (!s.ok()) { + LOG(ERROR) << "Failed to create session: " << s; + return nullptr; + } + return out_session; } Status NewSession(const SessionOptions& options, Session** out_session) { SessionFactory* factory; - const Status s = SessionFactory::GetFactory(options, &factory); + Status s = SessionFactory::GetFactory(options, &factory); if (!s.ok()) { *out_session = nullptr; LOG(ERROR) << s; return s; } - *out_session = factory->NewSession(options); - if (!*out_session) { - return errors::Internal("Failed to create session."); + s = factory->NewSession(options, out_session); + if (!s.ok()) { + *out_session = nullptr; } - return Status::OK(); + return s; } Status Reset(const SessionOptions& options, |