diff options
Diffstat (limited to 'tensorflow_lite_support/java/src/native/task/text/nlclassifier/bert_nlclassifier/bert_nl_classifier_jni.cc')
-rw-r--r-- | tensorflow_lite_support/java/src/native/task/text/nlclassifier/bert_nlclassifier/bert_nl_classifier_jni.cc | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/tensorflow_lite_support/java/src/native/task/text/nlclassifier/bert_nlclassifier/bert_nl_classifier_jni.cc b/tensorflow_lite_support/java/src/native/task/text/nlclassifier/bert_nlclassifier/bert_nl_classifier_jni.cc index 0866764e..aef82408 100644 --- a/tensorflow_lite_support/java/src/native/task/text/nlclassifier/bert_nlclassifier/bert_nl_classifier_jni.cc +++ b/tensorflow_lite_support/java/src/native/task/text/nlclassifier/bert_nlclassifier/bert_nl_classifier_jni.cc @@ -27,7 +27,8 @@ using ::tflite::support::utils::kInvalidPointer; using ::tflite::support::utils::ThrowException; using ::tflite::task::text::nlclassifier::BertNLClassifier; using ::tflite::task::text::nlclassifier::RunClassifier; -using ::tflite::task::text::nlclassifier::GetVersionNative; +using ::tflite::task::text::nlclassifier::GetModelVersionNative; +using ::tflite::task::text::nlclassifier::GetLabelsVersionNative; extern "C" JNIEXPORT void JNICALL Java_org_tensorflow_lite_task_text_nlclassifier_BertNLClassifier_deinitJni( @@ -73,9 +74,15 @@ Java_org_tensorflow_lite_task_text_nlclassifier_BertNLClassifier_classifyNative( } extern "C" JNIEXPORT jstring JNICALL -Java_org_tensorflow_lite_task_text_nlclassifier_BertNLClassifier_getVersionNative( +Java_org_tensorflow_lite_task_text_nlclassifier_BertNLClassifier_getModelVersionNative( JNIEnv* env, jclass clazz, jlong native_handle) { - return GetVersionNative(env, native_handle); + return GetModelVersionNative(env, native_handle); +} + +extern "C" JNIEXPORT jstring JNICALL +Java_org_tensorflow_lite_task_text_nlclassifier_BertNLClassifier_getLabelsVersionNative( + JNIEnv* env, jclass clazz, jlong native_handle) { + return GetLabelsVersionNative(env, native_handle); } } // namespace |