Project import generated by Copybara.

GitOrigin-RevId: 373e3ac1e5839befd95bf7d73ceff3c5f1171969

mediapipe/util/tflite/BUILD

@@ -34,10 +34,12 @@ cc_library(
     hdrs = ["cpu_op_resolver.h"],
     visibility = ["//visibility:public"],
     deps = [
-        "//mediapipe/framework:calculator_framework",
         "//mediapipe/framework/port:logging",
+        "//mediapipe/util/tflite/operations:landmarks_to_transform_matrix",
         "//mediapipe/util/tflite/operations:max_pool_argmax",
         "//mediapipe/util/tflite/operations:max_unpooling",
+        "//mediapipe/util/tflite/operations:transform_landmarks",
+        "//mediapipe/util/tflite/operations:transform_tensor_bilinear",
         "//mediapipe/util/tflite/operations:transpose_conv_bias",
         "@org_tensorflow//tensorflow/lite:builtin_op_data",
         "@org_tensorflow//tensorflow/lite:framework",

mediapipe/util/tflite/cpu_op_resolver.cc

@@ -15,8 +15,11 @@
 #include "mediapipe/util/tflite/cpu_op_resolver.h"
 
 #include "mediapipe/framework/port/logging.h"
+#include "mediapipe/util/tflite/operations/landmarks_to_transform_matrix.h"
 #include "mediapipe/util/tflite/operations/max_pool_argmax.h"
 #include "mediapipe/util/tflite/operations/max_unpooling.h"
+#include "mediapipe/util/tflite/operations/transform_landmarks.h"
+#include "mediapipe/util/tflite/operations/transform_tensor_bilinear.h"
 #include "mediapipe/util/tflite/operations/transpose_conv_bias.h"
 #include "tensorflow/lite/builtin_op_data.h"
 #include "tensorflow/lite/mutable_op_resolver.h"
@@ -31,6 +34,16 @@ void MediaPipe_RegisterTfLiteOpResolver(tflite::MutableOpResolver *resolver) {
                       tflite_operations::RegisterMaxUnpooling2D());
   resolver->AddCustom("Convolution2DTransposeBias",
                       tflite_operations::RegisterConvolution2DTransposeBias());
+
+  resolver->AddCustom("TransformTensorBilinear",
+                      tflite_operations::RegisterTransformTensorBilinearV2(),
+                      /*version=*/2);
+  resolver->AddCustom("TransformLandmarks",
+                      tflite_operations::RegisterTransformLandmarksV2(),
+                      /*version=*/2);
+  resolver->AddCustom("Landmarks2TransformMatrix",
+                      tflite_operations::RegisterLandmarksToTransformMatrixV2(),
+                      /*version=*/2);
 }
 
 }  // namespace mediapipe

mediapipe/util/tflite/operations/BUILD

@@ -22,6 +22,24 @@ package(default_visibility = [
     "//learning/brain/models/app_benchmarks/camera_models:__subpackages__",
 ])
 
+CUSTOM_OPS_PACKAGE = "@org_tensorflow//tensorflow/lite/delegates/gpu/common/mediapipe"
+
+cc_library(
+    name = "landmarks_to_transform_matrix",
+    srcs = ["landmarks_to_transform_matrix.cc"],
+    hdrs = ["landmarks_to_transform_matrix.h"],
+    deps = [
+        "@org_tensorflow//tensorflow/lite/delegates/gpu/common:types",
+        "@org_tensorflow//tensorflow/lite/kernels:kernel_util",
+        "@org_tensorflow//tensorflow/lite/kernels:padding",
+        "@org_tensorflow//tensorflow/lite/kernels/internal:common",
+        "@org_tensorflow//tensorflow/lite/kernels/internal:compatibility",
+        "@org_tensorflow//tensorflow/lite/kernels/internal:tensor",
+        "@org_tensorflow//tensorflow/lite/schema:schema_fbs",
+        CUSTOM_OPS_PACKAGE + ":landmarks_to_transform_matrix",
+    ],
+)
+
 cc_library(
     name = "max_pool_argmax",
     srcs = ["max_pool_argmax.cc"],
@@ -48,6 +66,38 @@ cc_library(
     ],
 )
 
+cc_library(
+    name = "transform_landmarks",
+    srcs = ["transform_landmarks.cc"],
+    hdrs = ["transform_landmarks.h"],
+    deps = [
+        "@org_tensorflow//tensorflow/lite/delegates/gpu/common:types",
+        "@org_tensorflow//tensorflow/lite/kernels:kernel_util",
+        "@org_tensorflow//tensorflow/lite/kernels:padding",
+        "@org_tensorflow//tensorflow/lite/kernels/internal:common",
+        "@org_tensorflow//tensorflow/lite/kernels/internal:compatibility",
+        "@org_tensorflow//tensorflow/lite/kernels/internal:tensor",
+        "@org_tensorflow//tensorflow/lite/schema:schema_fbs",
+        CUSTOM_OPS_PACKAGE + ":transform_landmarks",
+    ],
+)
+
+cc_library(
+    name = "transform_tensor_bilinear",
+    srcs = ["transform_tensor_bilinear.cc"],
+    hdrs = ["transform_tensor_bilinear.h"],
+    deps = [
+        "@org_tensorflow//tensorflow/lite/delegates/gpu/common:types",
+        "@org_tensorflow//tensorflow/lite/kernels:kernel_util",
+        "@org_tensorflow//tensorflow/lite/kernels:padding",
+        "@org_tensorflow//tensorflow/lite/kernels/internal:common",
+        "@org_tensorflow//tensorflow/lite/kernels/internal:compatibility",
+        "@org_tensorflow//tensorflow/lite/kernels/internal:tensor",
+        "@org_tensorflow//tensorflow/lite/schema:schema_fbs",
+        CUSTOM_OPS_PACKAGE + ":transform_tensor_bilinear",
+    ],
+)
+
 cc_library(
     name = "transpose_conv_bias",
     srcs = ["transpose_conv_bias.cc"],