Add tensorflow_text ops to TF mlir adapter

PiperOrigin-RevId: 641019302
google-ai-edge · Jun 6, 2024 · cc4095f · cc4095f
1 parent 057662c
commit cc4095f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/builtin-adapter/BUILD b/src/builtin-adapter/BUILD
@@ -27,6 +27,7 @@ cc_library(
         "//tools:attribute_printer",
         "//tools:load_opdefs",
         "@org_tensorflow//tensorflow/compiler/mlir/tensorflow:tensorflow_types",
+        "@org_tensorflow_text//tensorflow_text:ops_lib",
     ],
 )
 

diff --git a/src/builtin-adapter/WORKSPACE b/src/builtin-adapter/WORKSPACE
@@ -27,6 +27,15 @@ http_archive(
     ],
 )
 
+http_archive(
+    name = "org_tensorflow_text",
+    sha256 = "ccfb6af97705a9cb34b3a94bc46ec8fac88cfd6130997e86d15b3c49de9662eb",
+    strip_prefix = "text-2.16.1",
+    urls = [
+        "https://github.com/tensorflow/text/archive/v2.16.1.zip",
+    ],
+)
+
 http_archive(
     name = "rules_python",
     sha256 = "9d04041ac92a0985e344235f5d946f71ac543f1b1565f2cdbc9a2aaee8adf55b",