No public description

PiperOrigin-RevId: 686307894
google-ai-edge · Oct 16, 2024 · 91b56d7 · 91b56d7
1 parent c05adc5
commit 91b56d7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/mediapipe/tasks/cc/vision/image_segmenter/calculators/BUILD b/mediapipe/tasks/cc/vision/image_segmenter/calculators/BUILD
@@ -29,6 +29,13 @@ mediapipe_proto_library(
     ],
 )
 
+config_setting(
+    name = "force_gpu_segmentation",
+    values = {
+        "define": "mediapipe_force_gpu_segmentation=1",
+    },
+)
+
 cc_library(
     name = "tensors_to_segmentation_calculator",
     srcs = ["tensors_to_segmentation_calculator.cc"],