cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

No public description

Browse Source 
PiperOrigin-RevId: 574913082
This commit is contained in:
Fergus Henderson 2023-10-19 10:08:09 -07:00 committed by Copybara-Service
parent 7dca7ad24e
commit f9fa7cfbeb
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/util/tflite/BUILD
View File

@ -129,7 +129,9 @@ cc_library_with_tflite(
name = "tflite_model_loader", name = "tflite_model_loader",
srcs = ["tflite_model_loader.cc"], srcs = ["tflite_model_loader.cc"],
hdrs = ["tflite_model_loader.h"], hdrs = ["tflite_model_loader.h"],
tflite_deps = ["@org_tensorflow//tensorflow/lite:framework_stable"], tflite_deps = [
"@org_tensorflow//tensorflow/lite:framework_stable",
],
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//mediapipe/framework/api2:packet", "//mediapipe/framework/api2:packet",

1
mediapipe/util/tflite/tflite_model_loader.h
View File

@ -25,6 +25,7 @@
#include "tensorflow/lite/model.h" #include "tensorflow/lite/model.h"
namespace mediapipe { namespace mediapipe {
// Represents a TfLite model as a FlatBuffer. // Represents a TfLite model as a FlatBuffer.
using TfLiteModelPtr = using TfLiteModelPtr =
std::unique_ptr<tflite::FlatBufferModel, std::unique_ptr<tflite::FlatBufferModel,