Depend on shim targets in the Java API

PiperOrigin-RevId: 358370930
Change-Id: I1890368dec43716c2bbcbdc27b60a13e39903d06
This commit is contained in:
A. Unique TensorFlower 2021-02-19 03:16:57 -08:00 committed by TensorFlower Gardener
parent 47c5329897
commit 2a52449898
4 changed files with 6 additions and 9 deletions

View File

@ -751,7 +751,7 @@ cc_library(
deps = [
":mutable_op_resolver",
":op_resolver",
"//tensorflow/lite/core/shims:builtin_ops",
"//tensorflow/lite/kernels:builtin_ops",
],
)

View File

@ -21,8 +21,6 @@ namespace tflite_shims {
namespace ops {
namespace builtin {
using BuiltinOpResolver = ::tflite::ops::builtin::BuiltinOpResolver;
using BuiltinOpResolverWithoutDefaultDelegates =
::tflite::ops::builtin::BuiltinOpResolverWithoutDefaultDelegates;
} // namespace builtin
} // namespace ops
} // namespace tflite_shims

View File

@ -15,8 +15,8 @@ limitations under the License.
#include <memory>
#include "tensorflow/lite/core/shims/cc/kernels/register.h"
#include "tensorflow/lite/create_op_resolver.h"
#include "tensorflow/lite/kernels/register.h"
namespace tflite {
@ -26,9 +26,8 @@ namespace tflite {
// linking this in, and should provide a CreateOpResolver() with selected ops
// instead.
std::unique_ptr<MutableOpResolver> CreateOpResolver() { // NOLINT
return std::unique_ptr<::tflite_shims::ops::builtin::BuiltinOpResolver>(
new ::tflite_shims::ops::builtin::
BuiltinOpResolverWithoutDefaultDelegates());
return std::unique_ptr<tflite::ops::builtin::BuiltinOpResolver>(
new tflite::ops::builtin::BuiltinOpResolverWithoutDefaultDelegates());
}
} // namespace tflite

View File

@ -467,8 +467,8 @@ tflite_jni_binary(
deps = [
# Note that we explicitly include the C API here for convenience, as it
# allows bundling of the C lib w/ AAR distribution.
"//tensorflow/lite/core/shims:c_api",
"//tensorflow/lite/core/shims:c_api_experimental",
"//tensorflow/lite/c:c_api",
"//tensorflow/lite/c:c_api_experimental",
"//tensorflow/lite/delegates/nnapi/java/src/main/native",
"//tensorflow/lite/delegates/xnnpack:xnnpack_delegate",
"//tensorflow/lite/java/src/main/native",