From 90afe8ea3722babd7a59a1dcc382116bc5ee9682 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Tue, 16 Feb 2021 14:16:36 -0800 Subject: [PATCH] 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 --- .../lite/experimental/microfrontend/BUILD | 28 +++++++++++++++++-- 1 file changed, 25 insertions(+), 3 deletions(-) diff --git a/tensorflow/lite/experimental/microfrontend/BUILD b/tensorflow/lite/experimental/microfrontend/BUILD index 3f34902f1e6..9fc6cac31d0 100644 --- a/tensorflow/lite/experimental/microfrontend/BUILD +++ b/tensorflow/lite/experimental/microfrontend/BUILD @@ -2,9 +2,11 @@ load( "//tensorflow:tensorflow.bzl", + "tf_copts", "tf_custom_op_library", "tf_gen_op_libs", "tf_gen_op_wrapper_py", + "tf_opts_nortti_if_android", "tf_py_test", ) load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library") @@ -30,11 +32,31 @@ cc_library( cc_library( name = "audio_microfrontend_op_lib", 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 = [ - "//tensorflow/core:framework", - "//tensorflow/core:lib", "//tensorflow/lite/experimental/microfrontend/lib:frontend", - ], + ] + select({ + "//tensorflow:android": [ + "//tensorflow/core:portable_tensorflow_lib_lite", + ], + "//conditions:default": [ + "//tensorflow/core:framework", + "//tensorflow/core:lib", + ], + }), alwayslink = 1, )