talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update MediaPipe TFLite code to use generic "shim" symbols and headers

Browse Source 
PiperOrigin-RevId: 511265313
This commit is contained in:
MediaPipe Team
2023-02-21 11:58:32 -08:00
committed by Copybara-Service
parent 8a0830ce7b
commit 9a6d5e0444
6 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/calculators/tflite/BUILD
+1 -1
View File
@@ -320,7 +320,7 @@ cc_library(
"//mediapipe/framework:packet",
"//mediapipe/framework/port:ret_check",
"@com_google_absl//absl/status",
"@org_tensorflow//tensorflow/lite:framework",
"@org_tensorflow//tensorflow/lite/core/shims:framework_stable",
],
alwayslink = 1,
)
mediapipe/calculators/tflite/tflite_custom_op_resolver_calculator.cc
+2 -2
View File
@@ -66,7 +66,7 @@ class TfLiteCustomOpResolverCalculator : public CalculatorBase {
} else {
cc->OutputSidePackets()
.Index(0)
.Set<tflite::ops::builtin::BuiltinOpResolver>();
.Set<tflite_shims::ops::builtin::BuiltinOpResolver>();
}
return absl::OkStatus();
}
@@ -77,7 +77,7 @@ class TfLiteCustomOpResolverCalculator : public CalculatorBase {
const TfLiteCustomOpResolverCalculatorOptions& options =
cc->Options<TfLiteCustomOpResolverCalculatorOptions>();
std::unique_ptr<tflite::ops::builtin::BuiltinOpResolver> op_resolver;
std::unique_ptr<tflite_shims::ops::builtin::BuiltinOpResolver> op_resolver;
if (options.use_gpu()) {
op_resolver = absl::make_unique<mediapipe::OpResolver>();
} else {
mediapipe/calculators/tflite/tflite_model_calculator.cc
+1 -1
View File
@@ -21,7 +21,7 @@
#include "mediapipe/framework/packet.h"
#include "mediapipe/framework/port/ret_check.h"
#include "tensorflow/lite/allocation.h"
#include "tensorflow/lite/model.h"
#include "tensorflow/lite/core/shims/cc/model.h"
namespace mediapipe {