diff --git a/mediapipe/util/tflite/BUILD b/mediapipe/util/tflite/BUILD index 350ac230cf..b34d0e080f 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 65bd9ba728..f1a021f4de 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