aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/common_runtime/session.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/common_runtime/session.cc')
-rw-r--r--tensorflow/core/common_runtime/session.cc20
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,