Switch ARM64 builds to TFLite

Fixes mozilla/DeepSpeech#2676
mozilla · Jan 17, 2020 · 917d341 · 917d341
1 parent bd115ee
commit 917d341
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/.bazelrc b/.bazelrc
@@ -104,6 +104,7 @@ build:rpi3-armv8 --crosstool_top=//tools/arm_compiler/linaro-gcc72-aarch64:toolc
 build:rpi3-armv8 --cpu=armv8 --define=target_system=rpi3-armv8
 build:rpi3-armv8 --copt=-march=armv8-a --copt=-mtune=cortex-a53 --copt=-DRASPBERRY_PI --copt=-D_GLIBCXX_USE_CXX11_ABI=0 --copt=-std=gnu99
 build:rpi3-armv8 --define=tensorflow_mkldnn_contraction_kernel=0
+build:rpi3-armv8 --define=runtime=tflite
 build:rpi3-armv8_opt --copt=-funsafe-math-optimizations --copt=-ftree-vectorize --copt=-pipe
 
 # Options extracted from configure script

diff --git a/tensorflow/lite/kernels/internal/BUILD b/tensorflow/lite/kernels/internal/BUILD
@@ -101,6 +101,13 @@ config_setting(
     },
 )
 
+config_setting(
+    name = "rpi3-armv8",
+    define_values = {
+        "target_system": "rpi3-armv8"
+    },
+)
+
 config_setting(
     name = "armeabi-v7a",
     values = {
@@ -642,6 +649,9 @@ cc_library(
         ":rpi3": [
             ":neon_tensor_utils",
         ],
+        ":rpi3-armv8": [
+            ":neon_tensor_utils",
+        ],
         ":haswell": [
             ":sse_tensor_utils",
         ],