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/*", "models/testdata/*",
]) + [ ]) + [
"create_op_resolver.h", "create_op_resolver.h",
"create_op_resolver_with_selected_ops.cc",
]) ])
config_setting( 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( cc_test(
name = "util_test", name = "util_test",
size = "small", size = "small",

View File

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