Skip to content

Commit 706a9e2

Browse files
committed
[JNI] Fixed a Linux-related issue.
1 parent 895e1ea commit 706a9e2

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

api/src/main/scala/org/platanios/tensorflow/api/core/client/Session.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,7 @@ class Session private[api](
120120
(implicit executable: Executable[E], fetchable: Fetchable.Aux[F, R]): (R, Option[RunMetadata]) = {
121121
if (nativeHandle == 0)
122122
throw new IllegalStateException("This session has already been closed.")
123+
// TODO: !!! [JNI] Add a call to 'extend' once some JNI issues are resolved.
123124
val (inputs, inputTensors) = feeds.values.toSeq.unzip
124125
val inputTensorHandles: Array[Long] = inputTensors.map(_.resolve()).toArray
125126
val inputOpHandles: Array[Long] = inputs.map(_.op.nativeHandle).toArray

jni/src/main/native/include/tensorflow/c/python_api.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ limitations under the License.
1717

1818
#include "c_api.h"
1919
#include "c_api_internal.h"
20-
//#include "tensorflow/core/graph/validate.h"
2120

2221
namespace tensorflow {
2322

@@ -118,7 +117,7 @@ void SetAttr(
118117
void ExtendSession(TF_Session* session, TF_Status* status) {
119118
// TODO: !!! [JNI] Fix this.
120119
// ExtendSessionGraphHelper(session, status);
121-
session->extend_before_run = false;
120+
// session->extend_before_run = false;
122121
}
123122

124123
//// TODO(josh11b,mrry): Change Session to be able to use a Graph*

0 commit comments

Comments
 (0)