18 lines
898 B
Diff
18 lines
898 B
Diff
diff --git third_party/tflite/features.gni third_party/tflite/features.gni
|
|
index 2466577542552..d94d80cda02d6 100644
|
|
--- third_party/tflite/features.gni
|
|
+++ third_party/tflite/features.gni
|
|
@@ -7,9 +7,9 @@ import("//build/config/chrome_build.gni")
|
|
declare_args() {
|
|
# This enables building TFLite with XNNPACK. Currently only available for
|
|
# Linux, macOS and Windows arm64/x64/x86 targets and ChromeOS non-ARM targets.
|
|
- build_tflite_with_xnnpack =
|
|
- is_win || is_mac || is_linux ||
|
|
- (is_chromeos && target_cpu != "arm64" && target_cpu != "arm")
|
|
+ build_tflite_with_xnnpack = (is_win || is_mac || is_linux ||
|
|
+ (is_chromeos && current_cpu != "arm64" && current_cpu != "arm")) &&
|
|
+ (current_cpu == "arm64" || current_cpu == "x64" || current_cpu == "x86")
|
|
|
|
# Turns on TFLITE_WITH_RUY, using ruy as the gemm backend instead of gemmlowp.
|
|
build_tflite_with_ruy = true
|