diff options
author | Pete Warden <petewarden@google.com> | 2018-09-07 17:36:59 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-07 17:40:10 -0700 |
commit | 9982fd6c8831cbd2f58954f79ea71f26660393bc (patch) | |
tree | 108907bde953d0d70ee5d3b8323a99bb9b681563 /tensorflow/contrib/lite/java/src | |
parent | edda5e39e4e93ba60e4d31b6ecb1c295dead29c8 (diff) |
Modularize TF Lite interface definitions and reorganize file structure
PiperOrigin-RevId: 212064501
Diffstat (limited to 'tensorflow/contrib/lite/java/src')
-rw-r--r-- | tensorflow/contrib/lite/java/src/main/native/nativeinterpreterwrapper_jni.h | 8 | ||||
-rw-r--r-- | tensorflow/contrib/lite/java/src/main/native/tensor_jni.h | 2 |
2 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/contrib/lite/java/src/main/native/nativeinterpreterwrapper_jni.h b/tensorflow/contrib/lite/java/src/main/native/nativeinterpreterwrapper_jni.h index 55ca47fed7..06b35d77c8 100644 --- a/tensorflow/contrib/lite/java/src/main/native/nativeinterpreterwrapper_jni.h +++ b/tensorflow/contrib/lite/java/src/main/native/nativeinterpreterwrapper_jni.h @@ -20,7 +20,7 @@ limitations under the License. #include <stdio.h> #include <time.h> #include <vector> -#include "tensorflow/contrib/lite/context.h" +#include "tensorflow/contrib/lite/c/c_api_internal.h" #include "tensorflow/contrib/lite/interpreter.h" #include "tensorflow/contrib/lite/java/src/main/native/exception_jni.h" #include "tensorflow/contrib/lite/java/src/main/native/tensor_jni.h" @@ -124,9 +124,9 @@ Java_org_tensorflow_lite_NativeInterpreterWrapper_useNNAPI(JNIEnv* env, */ JNIEXPORT void JNICALL Java_org_tensorflow_lite_NativeInterpreterWrapper_numThreads(JNIEnv* env, - jclass clazz, - jlong handle, - jint num_threads); + jclass clazz, + jlong handle, + jint num_threads); /* * Class: org_tensorflow_lite_NativeInterpreterWrapper * Method: diff --git a/tensorflow/contrib/lite/java/src/main/native/tensor_jni.h b/tensorflow/contrib/lite/java/src/main/native/tensor_jni.h index c020f13d9c..2f73128bdf 100644 --- a/tensorflow/contrib/lite/java/src/main/native/tensor_jni.h +++ b/tensorflow/contrib/lite/java/src/main/native/tensor_jni.h @@ -17,7 +17,7 @@ limitations under the License. #define TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_TENSOR_JNI_H_ #include <jni.h> -#include "tensorflow/contrib/lite/context.h" +#include "tensorflow/contrib/lite/c/c_api_internal.h" #ifdef __cplusplus extern "C" { |