From f9fa7cfbeb99b93a1625fb08bb1d5096d4704616 Mon Sep 17 00:00:00 2001 From: Fergus Henderson Date: Thu, 19 Oct 2023 10:08:09 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 574913082 --- mediapipe/util/tflite/BUILD | 4 +++- mediapipe/util/tflite/tflite_model_loader.h | 1 + 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/mediapipe/util/tflite/BUILD b/mediapipe/util/tflite/BUILD index 350ac230c..b34d0e080 100644 --- a/mediapipe/util/tflite/BUILD +++ b/mediapipe/util/tflite/BUILD @@ -129,7 +129,9 @@ cc_library_with_tflite( name = "tflite_model_loader", srcs = ["tflite_model_loader.cc"], hdrs = ["tflite_model_loader.h"], - tflite_deps = ["@org_tensorflow//tensorflow/lite:framework_stable"], + tflite_deps = [ + "@org_tensorflow//tensorflow/lite:framework_stable", + ], visibility = ["//visibility:public"], deps = [ "//mediapipe/framework/api2:packet", diff --git a/mediapipe/util/tflite/tflite_model_loader.h b/mediapipe/util/tflite/tflite_model_loader.h index 65bd9ba72..f1a021f4d 100644 --- a/mediapipe/util/tflite/tflite_model_loader.h +++ b/mediapipe/util/tflite/tflite_model_loader.h @@ -25,6 +25,7 @@ #include "tensorflow/lite/model.h" namespace mediapipe { + // Represents a TfLite model as a FlatBuffer. using TfLiteModelPtr = std::unique_ptr