diff options
author | Jared Duke <jdduke@google.com> | 2018-09-04 13:29:25 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-04 13:34:31 -0700 |
commit | bfdb7a408c1ea519df9f970220e36c89e8fe1cf3 (patch) | |
tree | 97a0ba9883a5135df98f125f315f4eb2240d08c8 /tensorflow/contrib/lite/kernels/BUILD | |
parent | c0a9c988f75082b0c8521c1343874c4ce9a10dd6 (diff) |
Disable rtti for builtin TFLite kernels
PiperOrigin-RevId: 211514002
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/kernels/BUILD | 20 |
1 files changed, 15 insertions, 5 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD index ca66fa6aa0..ab989c5425 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -6,7 +6,7 @@ licenses(["notice"]) # Apache 2.0 load("//tensorflow/contrib/lite:build_def.bzl", "tflite_copts") load("//tensorflow/contrib/lite:special_rules.bzl", "tflite_portable_test_suite") -load("//tensorflow:tensorflow.bzl", "tf_cc_test") +load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_opts_nortti_if_android") # Suppress warnings that are introduced by Eigen Tensor. EXTRA_EIGEN_COPTS = select({ @@ -147,7 +147,7 @@ tf_cc_test( ) cc_library( - name = "builtin_ops", + name = "builtin_op_kernels", srcs = [ "activations.cc", "add.cc", @@ -192,7 +192,6 @@ cc_library( "pooling.cc", "pow.cc", "reduce.cc", - "register.cc", "reshape.cc", "resize_bilinear.cc", "select.cc", @@ -217,9 +216,9 @@ cc_library( ], hdrs = [ "padding.h", - "register.h", ], - copts = tflite_copts() + EXTRA_EIGEN_COPTS, + copts = tflite_copts() + tf_opts_nortti_if_android() + EXTRA_EIGEN_COPTS, + visibility = ["//visibility:private"], deps = [ ":activation_functor", ":eigen_support", @@ -243,6 +242,17 @@ cc_library( ], ) +cc_library( + name = "builtin_ops", + srcs = ["register.cc"], + hdrs = ["register.h"], + deps = [ + ":builtin_op_kernels", + "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite:util", + ], +) + tf_cc_test( name = "audio_spectrogram_test", size = "small", |