Fixed backward reference error of tflite_custom_android_library

PiperOrigin-RevId: 346660735
Change-Id: Iff7a55bce35a13a28d57b11f995487e8710d0113
This commit is contained in:
Thai Nguyen 2020-12-09 16:17:08 -08:00 committed by TensorFlower Gardener
parent cc5ea84696
commit b48df5c1f3
2 changed files with 3 additions and 17 deletions

View File

@ -16,6 +16,7 @@ exports_files(glob([
"models/testdata/*",
]) + [
"create_op_resolver.h",
"create_op_resolver_with_selected_ops.cc",
])
config_setting(
@ -722,21 +723,6 @@ cc_library(
],
)
# This target is created for tflite_custom_cc_library build rule. It requires
# the header file generated from gen_selected_ops so should not be depended on
# directly.
# TODO(b/174972014): Generate this target to give RegisterSelectedOps a custom namespace.
cc_library(
name = "create_op_resolver_with_selected_ops",
srcs = ["create_op_resolver_with_selected_ops.cc"],
hdrs = ["create_op_resolver.h"],
copts = tflite_copts(),
deps = [
"//tensorflow/lite:mutable_op_resolver",
"//tensorflow/lite:op_resolver",
],
)
cc_test(
name = "util_test",
size = "small",

View File

@ -800,7 +800,7 @@ def tflite_custom_cc_library(
model = models,
)
real_srcs.append(":%s_registration" % name)
real_deps.append("//tensorflow/lite:create_op_resolver_with_selected_ops")
real_srcs.append("//tensorflow/lite:create_op_resolver_with_selected_ops.cc")
else:
# Support all operators if `models` not specified.
real_deps.append("//tensorflow/lite/java/src/main/native")
@ -855,8 +855,8 @@ def tflite_custom_android_library(
name = "libtensorflowlite_jni.so",
linkscript = "//tensorflow/lite/java:tflite_version_script.lds",
deps = [
":%s_cc" % name,
"//tensorflow/lite/java/src/main/native:native_framework_only",
":%s_cc" % name, # Placed below native_framework_only to avoid backward reference error.
],
)