From 91b56d756315bcfa07ebedc63c476cb850050f93 Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Tue, 15 Oct 2024 18:20:04 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 686307894 --- .../tasks/cc/vision/image_segmenter/calculators/BUILD | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/mediapipe/tasks/cc/vision/image_segmenter/calculators/BUILD b/mediapipe/tasks/cc/vision/image_segmenter/calculators/BUILD index 93a49fdf83..4b8176cfed 100644 --- 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"],