aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMiao Wang <miaowang@google.com>2017-12-14 15:30:54 -0800
committerMiao Wang <miaowang@google.com>2017-12-14 15:30:54 -0800
commita07e14cf255c2d4f29733ac25a72f3166230a0f6 (patch)
tree5cb77f9866e91833834e756cb08bdb777b5fac1b
parentf70908d9b0bcd74fa483be183838353e293bde97 (diff)
downloadtensorflow-temp_72223856.tar.gz
Update TFLite build file to provide dependencies for NN API.temp_72223856
Test: mm Change-Id: Iae3eff80bc0c0d98960d9d7c7156691cf6486066
-rw-r--r--Android.bp1
-rw-r--r--tensorflow/contrib/lite/kernels/Android.bp13
2 files changed, 14 insertions, 0 deletions
diff --git a/Android.bp b/Android.bp
index 242c397baa3..3dc4640f7e9 100644
--- a/Android.bp
+++ b/Android.bp
@@ -15,6 +15,7 @@
cc_library_headers {
name: "tensorflow_headers",
export_include_dirs: ["."],
+ vendor_available: true,
}
subdirs = ["tensorflow/contrib/lite"]
diff --git a/tensorflow/contrib/lite/kernels/Android.bp b/tensorflow/contrib/lite/kernels/Android.bp
index 391aa234ff0..f077bcfbed9 100644
--- a/tensorflow/contrib/lite/kernels/Android.bp
+++ b/tensorflow/contrib/lite/kernels/Android.bp
@@ -13,6 +13,19 @@
// limitations under the License.
cc_library_static {
+ name: "libtflite_kernel_utils",
+ defaults: ["tflite_defaults"],
+ vendor_available: true,
+ srcs: [
+ "kernel_util.cc",
+ "internal/tensor_utils.cc",
+ "internal/quantization_util.cc",
+ "internal/reference/portable_tensor_utils.cc",
+ "internal/optimized/neon_tensor_utils.cc",
+ ],
+}
+
+cc_library_static {
name: "libtflite_kernels",
defaults: ["tflite_defaults"],
srcs: [