diff options
author | 2017-03-10 15:13:44 -0800 | |
---|---|---|
committer | 2017-03-10 15:29:45 -0800 | |
commit | c67cfdd2fb0a01293df130d1ba955e63bec158af (patch) | |
tree | 26c732f2f65eed9ef794d64a473d048f727e2861 /tensorflow/java/src/main/native/operation_jni.h | |
parent | 81cd89806576ec592c00aa446f14ef15480d5dff (diff) |
Java: Add a method to retrieve the DataType of an Output.
Change: 149803866
Diffstat (limited to 'tensorflow/java/src/main/native/operation_jni.h')
-rw-r--r-- | tensorflow/java/src/main/native/operation_jni.h | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/java/src/main/native/operation_jni.h b/tensorflow/java/src/main/native/operation_jni.h index 152ab139a5..fe14882dde 100644 --- a/tensorflow/java/src/main/native/operation_jni.h +++ b/tensorflow/java/src/main/native/operation_jni.h @@ -56,6 +56,14 @@ JNIEXPORT jlongArray JNICALL Java_org_tensorflow_Operation_shape(JNIEnv *, jclass, jlong, jlong, jint); +/* + * Class: org_tensorflow_Operation + * Method: dtype + * Signature: (JJI)I + */ +JNIEXPORT jint JNICALL Java_org_tensorflow_Operation_dtype(JNIEnv *, jclass, + jlong, jlong, jint); + #ifdef __cplusplus } // extern "C" #endif // __cplusplus |