From abb6b8e73cc0c7b566536df52e93a7eb6327b788 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 12 Oct 2017 15:08:11 -0700
Subject: [PATCH] Add linux_arm64 and linux_armhf config_setting.

PiperOrigin-RevId: 172018709
---
 tensorflow/BUILD                      | 16 ++++++++++++++++
 tensorflow/compiler/aot/tfcompile.bzl |  2 ++
 tensorflow/tensorflow.bzl             |  6 ++++++
 3 files changed, 24 insertions(+)

diff --git a/tensorflow/BUILD b/tensorflow/BUILD
index 64758dee0ee..9d07697d012 100644
--- a/tensorflow/BUILD
+++ b/tensorflow/BUILD
@@ -141,6 +141,22 @@ config_setting(
     visibility = ["//visibility:public"],
 )
 
+config_setting(
+    name = "linux_armhf",
+    values = {
+        "cpu": "armeabi-v7a",
+    },
+    visibility = ["//visibility:public"],
+)
+
+config_setting(
+    name = "linux_arm64",
+    values = {
+        "cpu": "arm64-v8a",
+    },
+    visibility = ["//visibility:public"],
+)
+
 config_setting(
     name = "debug",
     values = {
diff --git a/tensorflow/compiler/aot/tfcompile.bzl b/tensorflow/compiler/aot/tfcompile.bzl
index 4888760acd4..c900d201d29 100644
--- a/tensorflow/compiler/aot/tfcompile.bzl
+++ b/tensorflow/compiler/aot/tfcompile.bzl
@@ -302,6 +302,8 @@ def target_llvm_triple():
       "//tensorflow:android_arm": "armv7-none-android",
       "//tensorflow:android_arm64": "aarch64-none-android",
       "//tensorflow:android_x86": "i686-none-android",
+      "//tensorflow:linux_armhf": "armv7-none-linux-gnueabihf",
+      "//tensorflow:linux_arm64": "aarch64-none-linux-gnu",
       "//tensorflow:linux_ppc64le": "ppc64le-ibm-linux-gnu",
       "//tensorflow:darwin": "x86_64-none-darwin",
       "//conditions:default": "x86_64-pc-linux",
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl
index 0f074151db2..5ec31e492ce 100644
--- a/tensorflow/tensorflow.bzl
+++ b/tensorflow/tensorflow.bzl
@@ -112,6 +112,9 @@ def if_ios(a):
 def if_mobile(a):
   return select({
       clean_dep("//tensorflow:android"): a,
+      # Treat arm linux devices as mobile.
+      clean_dep("//tensorflow:linux_arm64"): a,
+      clean_dep("//tensorflow:linux_armhf"): a,
       clean_dep("//tensorflow:ios"): a,
       "//conditions:default": [],
   })
@@ -120,6 +123,9 @@ def if_mobile(a):
 def if_not_mobile(a):
   return select({
       clean_dep("//tensorflow:android"): [],
+      # Treat arm linux devices as mobile.
+      clean_dep("//tensorflow:linux_arm64"): [],
+      clean_dep("//tensorflow:linux_armhf"): [],
       clean_dep("//tensorflow:ios"): [],
       "//conditions:default": a,
   })