diff options
Diffstat (limited to 'tensorflow_lite_support/java/src/native/task/text/qa/BUILD')
-rw-r--r-- | tensorflow_lite_support/java/src/native/task/text/qa/BUILD | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/tensorflow_lite_support/java/src/native/task/text/qa/BUILD b/tensorflow_lite_support/java/src/native/task/text/qa/BUILD new file mode 100644 index 00000000..9753e329 --- /dev/null +++ b/tensorflow_lite_support/java/src/native/task/text/qa/BUILD @@ -0,0 +1,30 @@ +load("@org_tensorflow//tensorflow/lite:build_def.bzl", "tflite_jni_binary") + +package( + default_visibility = ["//tensorflow_lite_support:users"], + licenses = ["notice"], # Apache 2.0 +) + +exports_files([ + "bert_question_answerer_jni.cc", +]) + +tflite_jni_binary( + name = "libtask_text_jni.so", + srcs = [ + "bert_question_answerer_jni.cc", + ], + linkscript = "//tensorflow_lite_support/java:default_version_script.lds", + deps = [ + "//tensorflow_lite_support/cc/task/text/qa:bert_question_answerer", + "//tensorflow_lite_support/cc/utils:jni_utils", + "//tensorflow_lite_support/java/jni", + ], +) + +cc_library( + name = "bert_question_answerer_native", + srcs = [ + ":libtask_text_jni.so", + ], +) |