Update MediaPipe TFLite code to use generic "shim" symbols and headers
PiperOrigin-RevId: 511265313
This commit is contained in:
parent
8a0830ce7b
commit
9a6d5e0444
|
@ -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,
|
||||
)
|
||||
|
|
|
@ -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 {
|
||||
|
|
|
@ -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 {
|
||||
|
||||
|
|
|
@ -44,8 +44,8 @@ cc_library(
|
|||
"//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",
|
||||
"@org_tensorflow//tensorflow/lite/kernels:builtin_ops",
|
||||
"@org_tensorflow//tensorflow/lite/core/shims:builtin_ops",
|
||||
"@org_tensorflow//tensorflow/lite/core/shims:framework_stable",
|
||||
],
|
||||
# For using the symbol `MediaPipe_RegisterTfLiteOpResolver` in Python
|
||||
# with `tensorflow.lite.python.interpreter.InterpreterWithCustomOps`.
|
||||
|
@ -69,7 +69,7 @@ cc_library(
|
|||
hdrs = ["op_resolver.h"],
|
||||
deps = [
|
||||
"@org_tensorflow//tensorflow/lite:builtin_op_data",
|
||||
"@org_tensorflow//tensorflow/lite/kernels:builtin_ops",
|
||||
"@org_tensorflow//tensorflow/lite/core/shims:builtin_ops",
|
||||
],
|
||||
)
|
||||
|
||||
|
|
|
@ -15,7 +15,7 @@
|
|||
#ifndef MEDIAPIPE_UTIL_TFLITE_CPU_OP_RESOLVER_H_
|
||||
#define MEDIAPIPE_UTIL_TFLITE_CPU_OP_RESOLVER_H_
|
||||
|
||||
#include "tensorflow/lite/kernels/register.h"
|
||||
#include "tensorflow/lite/core/shims/cc/kernels/register.h"
|
||||
|
||||
namespace mediapipe {
|
||||
|
||||
|
@ -27,8 +27,8 @@ extern "C" void MediaPipe_RegisterTfLiteOpResolver(tflite::MutableOpResolver*);
|
|||
|
||||
// This resolver is used for the custom ops introduced by
|
||||
// `MediaPipe_RegisterTfLiteOpResolver` (see above).
|
||||
class CpuOpResolver
|
||||
: public tflite::ops::builtin::BuiltinOpResolverWithoutDefaultDelegates {
|
||||
class CpuOpResolver : public tflite_shims::ops::builtin::
|
||||
BuiltinOpResolverWithoutDefaultDelegates {
|
||||
public:
|
||||
CpuOpResolver() { MediaPipe_RegisterTfLiteOpResolver(this); }
|
||||
};
|
||||
|
|
|
@ -15,13 +15,13 @@
|
|||
#ifndef MEDIAPIPE_UTIL_TFLITE_OP_RESOLVER_H_
|
||||
#define MEDIAPIPE_UTIL_TFLITE_OP_RESOLVER_H_
|
||||
|
||||
#include "tensorflow/lite/kernels/register.h"
|
||||
#include "tensorflow/lite/core/shims/cc/kernels/register.h"
|
||||
|
||||
namespace mediapipe {
|
||||
|
||||
// This OpResolver is used for supporting "Convolution2DTransposeBias" on GPU.
|
||||
class OpResolver
|
||||
: public tflite::ops::builtin::BuiltinOpResolverWithoutDefaultDelegates {
|
||||
class OpResolver : public tflite_shims::ops::builtin::
|
||||
BuiltinOpResolverWithoutDefaultDelegates {
|
||||
public:
|
||||
OpResolver();
|
||||
};
|
||||
|
|
Loading…
Reference in New Issue
Block a user