Selectively choose the dependencies of audio_microfrontend_op_lib for different target os so that we can use the op on android.
PiperOrigin-RevId: 357803057 Change-Id: I7c87b1f2b5a775fce1dcff719f595f902be84fda
This commit is contained in:
parent
632b2acccf
commit
90afe8ea37
@ -2,9 +2,11 @@
|
|||||||
|
|
||||||
load(
|
load(
|
||||||
"//tensorflow:tensorflow.bzl",
|
"//tensorflow:tensorflow.bzl",
|
||||||
|
"tf_copts",
|
||||||
"tf_custom_op_library",
|
"tf_custom_op_library",
|
||||||
"tf_gen_op_libs",
|
"tf_gen_op_libs",
|
||||||
"tf_gen_op_wrapper_py",
|
"tf_gen_op_wrapper_py",
|
||||||
|
"tf_opts_nortti_if_android",
|
||||||
"tf_py_test",
|
"tf_py_test",
|
||||||
)
|
)
|
||||||
load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library")
|
load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library")
|
||||||
@ -30,11 +32,31 @@ cc_library(
|
|||||||
cc_library(
|
cc_library(
|
||||||
name = "audio_microfrontend_op_lib",
|
name = "audio_microfrontend_op_lib",
|
||||||
srcs = ["ops/audio_microfrontend_op.cc"],
|
srcs = ["ops/audio_microfrontend_op.cc"],
|
||||||
|
copts = tf_copts(android_optimization_level_override = None) + tf_opts_nortti_if_android() + [
|
||||||
|
"-Wno-narrowing",
|
||||||
|
"-Wno-sign-compare",
|
||||||
|
"-Wno-overloaded-virtual",
|
||||||
|
] + select({
|
||||||
|
"//tensorflow:android": [
|
||||||
|
# Selective registration uses constexprs with recursive
|
||||||
|
# string comparisons; that can lead to compiler errors, so
|
||||||
|
# we increase the constexpr recursion depth.
|
||||||
|
"-fconstexpr-depth=1024",
|
||||||
|
"-Oz",
|
||||||
|
],
|
||||||
|
"//conditions:default": [],
|
||||||
|
}),
|
||||||
deps = [
|
deps = [
|
||||||
|
"//tensorflow/lite/experimental/microfrontend/lib:frontend",
|
||||||
|
] + select({
|
||||||
|
"//tensorflow:android": [
|
||||||
|
"//tensorflow/core:portable_tensorflow_lib_lite",
|
||||||
|
],
|
||||||
|
"//conditions:default": [
|
||||||
"//tensorflow/core:framework",
|
"//tensorflow/core:framework",
|
||||||
"//tensorflow/core:lib",
|
"//tensorflow/core:lib",
|
||||||
"//tensorflow/lite/experimental/microfrontend/lib:frontend",
|
|
||||||
],
|
],
|
||||||
|
}),
|
||||||
alwayslink = 1,
|
alwayslink = 1,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user