diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-10 15:45:46 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-10 15:46:09 -0700 |
commit | 3bff5431380a3c23f71ecc198af7ca9b176113dd (patch) | |
tree | 8a2b993cb20c1e441cff337f7ccdbb12ffbc8114 /tensorflow/java | |
parent | 274fa1db82bd5fc3c983cc7081a9dd80979f2013 (diff) | |
parent | 9608d3bf6636a4c139433bf500314105328e491c (diff) |
Merge pull request #20558 from weberxie:master
PiperOrigin-RevId: 204026198
Diffstat (limited to 'tensorflow/java')
-rw-r--r-- | tensorflow/java/src/main/native/session_jni.cc | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/tensorflow/java/src/main/native/session_jni.cc b/tensorflow/java/src/main/native/session_jni.cc index cb54daf137..8b11525785 100644 --- a/tensorflow/java/src/main/native/session_jni.cc +++ b/tensorflow/java/src/main/native/session_jni.cc @@ -86,20 +86,22 @@ JNIEXPORT jlong JNICALL Java_org_tensorflow_Session_allocate2( TF_Graph* graph = reinterpret_cast<TF_Graph*>(graph_handle); TF_Status* status = TF_NewStatus(); TF_SessionOptions* opts = TF_NewSessionOptions(); - const char* ctarget = nullptr; jbyte* cconfig = nullptr; - if (target != nullptr) { - ctarget = env->GetStringUTFChars(target, nullptr); - } if (config != nullptr) { cconfig = env->GetByteArrayElements(config, nullptr); TF_SetConfig(opts, cconfig, static_cast<size_t>(env->GetArrayLength(config)), status); if (!throwExceptionIfNotOK(env, status)) { env->ReleaseByteArrayElements(config, cconfig, JNI_ABORT); + TF_DeleteSessionOptions(opts); + TF_DeleteStatus(status); return 0; } } + const char* ctarget = nullptr; + if (target != nullptr) { + ctarget = env->GetStringUTFChars(target, nullptr); + } TF_Session* session = TF_NewSession(graph, opts, status); if (config != nullptr) { env->ReleaseByteArrayElements(config, cconfig, JNI_ABORT); |