No public description

PiperOrigin-RevId: 590950821
google-ai-edge · Dec 14, 2023 · df7fead · df7fead
1 parent 4004c2d
commit df7fead
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 40 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -1,6 +1,7 @@
 workspace(name = "mediapipe")
 
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+load("@//third_party:shared_dependencies.bzl", "mediapipe_absl", "mediapipe_sentencepiece", "mediapipe_flatbuffers", "mediapipe_tensorflow")
 
 # Protobuf expects an //external:python_headers target
 bind(
@@ -21,7 +22,6 @@ bazel_skylib_workspace()
 load("@bazel_skylib//lib:versions.bzl", "versions")
 versions.check(minimum_bazel_version = "3.7.2")
 
-load("@//third_party:shared_dependencies.bzl", "mediapipe_absl", "mediapipe_sentencepiece")
 mediapipe_absl()
 mediapipe_sentencepiece()
 
@@ -208,8 +208,7 @@ http_archive(
     urls = ["https://github.com/protocolbuffers/protobuf/archive/v3.19.1.tar.gz"],
 )
 
-load("@//third_party/flatbuffers:workspace.bzl", flatbuffers = "repo")
-flatbuffers()
+mediapipe_flatbuffers()
 
 http_archive(
     name = "com_google_audio_tools",
@@ -229,7 +228,6 @@ http_archive(
     build_file = "@//third_party:pffft.BUILD",
 )
 
-
 http_archive(
     name = "darts_clone",
     build_file = "@//third_party:darts_clone.BUILD",
@@ -463,42 +461,7 @@ maven_install(
     version_conflict_policy = "pinned",
 )
 
-# Needed by TensorFlow
-http_archive(
-    name = "io_bazel_rules_closure",
-    sha256 = "e0a111000aeed2051f29fcc7a3f83be3ad8c6c93c186e64beb1ad313f0c7f9f9",
-    strip_prefix = "rules_closure-cf1e44edb908e9616030cc83d085989b8e6cd6df",
-    urls = [
-        "http://mirror.tensorflow.org/github.com/bazelbuild/rules_closure/archive/cf1e44edb908e9616030cc83d085989b8e6cd6df.tar.gz",
-        "https://github.com/bazelbuild/rules_closure/archive/cf1e44edb908e9616030cc83d085989b8e6cd6df.tar.gz",  # 2019-04-04
-    ],
-)
-
-# TensorFlow repo should always go after the other external dependencies.
-# TF on 2023-07-26.
-_TENSORFLOW_GIT_COMMIT = "e92261fd4cec0b726692081c4d2966b75abf31dd"
-# curl -L https://github.com/tensorflow/tensorflow/archive/<TENSORFLOW_GIT_COMMIT>.tar.gz | shasum -a 256
-_TENSORFLOW_SHA256 = "478a229bd4ec70a5b568ac23b5ea013d9fca46a47d6c43e30365a0412b9febf4"
-http_archive(
-    name = "org_tensorflow",
-    urls = [
-      "https://github.com/tensorflow/tensorflow/archive/%s.tar.gz" % _TENSORFLOW_GIT_COMMIT,
-    ],
-    patches = [
-        "@//third_party:org_tensorflow_compatibility_fixes.diff",
-        "@//third_party:org_tensorflow_system_python.diff",
-        # Diff is generated with a script, don't update it manually.
-        "@//third_party:org_tensorflow_custom_ops.diff",
-        # Works around Bazel issue with objc_library.
-        # See https://github.com/bazelbuild/bazel/issues/19912
-        "@//third_party:org_tensorflow_objc_build_fixes.diff",
-    ],
-    patch_args = [
-        "-p1",
-    ],
-    strip_prefix = "tensorflow-%s" % _TENSORFLOW_GIT_COMMIT,
-    sha256 = _TENSORFLOW_SHA256,
-)
+mediapipe_tensorflow()
 
 load("@org_tensorflow//tensorflow:workspace3.bzl", "tf_workspace3")
 tf_workspace3()

diff --git a/third_party/shared_dependencies.bzl b/third_party/shared_dependencies.bzl
@@ -1,9 +1,18 @@
 """MediaPipe's shared dependencies that can be used by dependent projects. Includes build patches."""
 
+load("@//third_party/flatbuffers:workspace.bzl", flatbuffers = "repo")
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
+# TensorFlow repo should always go after the other external dependencies.
+# TF on 2023-07-26.
+_TENSORFLOW_GIT_COMMIT = "e92261fd4cec0b726692081c4d2966b75abf31dd"
+
+# curl -L https://github.com/tensorflow/tensorflow/archive/<TENSORFLOW_GIT_COMMIT>.tar.gz | shasum -a 256
+_TENSORFLOW_SHA256 = "478a229bd4ec70a5b568ac23b5ea013d9fca46a47d6c43e30365a0412b9febf4"
+
 # ABSL cpp library lts_2023_01_25.
 def mediapipe_absl():
+    """Exports the ABSL depedency on TensorFlow."""
     http_archive(
         name = "com_google_absl",
         urls = [
@@ -20,6 +29,7 @@ def mediapipe_absl():
     )
 
 def mediapipe_sentencepiece():
+    """Exports the Semtencepiece depedency on TensorFlow."""
     http_archive(
         name = "com_google_sentencepiece",
         strip_prefix = "sentencepiece-0.1.96",
@@ -31,3 +41,40 @@ def mediapipe_sentencepiece():
         patches = ["@//third_party:com_google_sentencepiece.diff"],
         patch_args = ["-p1"],
     )
+
+def mediapipe_flatbuffers():
+    """Exports the FlatBuffers depedency on TensorFlow."""
+    flatbuffers()
+
+def mediapipe_tensorflow():
+    """Exports the MediaPipe depedency on TensorFlow."""
+
+    # Needed by TensorFlow
+    http_archive(
+        name = "io_bazel_rules_closure",
+        sha256 = "e0a111000aeed2051f29fcc7a3f83be3ad8c6c93c186e64beb1ad313f0c7f9f9",
+        strip_prefix = "rules_closure-cf1e44edb908e9616030cc83d085989b8e6cd6df",
+        urls = [
+            "http://mirror.tensorflow.org/github.com/bazelbuild/rules_closure/archive/cf1e44edb908e9616030cc83d085989b8e6cd6df.tar.gz",
+            "https://github.com/bazelbuild/rules_closure/archive/cf1e44edb908e9616030cc83d085989b8e6cd6df.tar.gz",  # 2019-04-04
+        ],
+    )
+
+    http_archive(
+        name = "org_tensorflow",
+        urls = ["https://github.com/tensorflow/tensorflow/archive/%s.tar.gz" % _TENSORFLOW_GIT_COMMIT],
+        patches = [
+            "@//third_party:org_tensorflow_compatibility_fixes.diff",
+            "@//third_party:org_tensorflow_system_python.diff",
+            # Diff is generated with a script, don't update it manually.
+            "@//third_party:org_tensorflow_custom_ops.diff",
+            # Works around Bazel issue with objc_library.
+            # See https://github.com/bazelbuild/bazel/issues/19912
+            "@//third_party:org_tensorflow_objc_build_fixes.diff",
+        ],
+        patch_args = [
+            "-p1",
+        ],
+        strip_prefix = "tensorflow-%s" % _TENSORFLOW_GIT_COMMIT,
+        sha256 = _TENSORFLOW_SHA256,
+    )