diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 1ceefe3a423..8a0918b416f 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -260,36 +260,6 @@ config_setting( visibility = ["//visibility:public"], ) -config_setting( - name = "armeabi", - values = {"cpu": "armeabi"}, - visibility = ["//visibility:public"], -) - -config_setting( - name = "armeabi-v7a", - values = {"cpu": "armeabi-v7a"}, - visibility = ["//visibility:public"], -) - -config_setting( - name = "arm64-v8a", - values = {"cpu": "arm64-v8a"}, - visibility = ["//visibility:public"], -) - -selects.config_setting_group( - name = "arm_any", - match_any = [ - ":arm", - ":armeabi", - ":armeabi-v7a", - ":arm64-v8a", - ":linux_aarch64", - ":linux_armhf", - ], -) - config_setting( name = "freebsd", values = {"cpu": "freebsd"}, diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 0dee2a48e76..9ace481a991 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -817,9 +817,10 @@ cc_library( srcs = ["eigen_contraction_kernel.cc"], hdrs = ["eigen_contraction_kernel.h"], defines = select({ - "//tensorflow:android_x86": [], - "//tensorflow:arm_any": [], + "//tensorflow:android": [], + "//tensorflow:arm": [], "//tensorflow:ios": [], + "//tensorflow:linux_aarch64": [], "//tensorflow:linux_ppc64le": [], "//conditions:default": [ "TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL", @@ -831,9 +832,10 @@ cc_library( "//third_party/eigen3", "//tensorflow/core/platform:dynamic_annotations", ] + select({ - "//tensorflow:android_x86": [], - "//tensorflow:arm_any": [], + "//tensorflow:android": [], + "//tensorflow:arm": [], "//tensorflow:ios": [], + "//tensorflow:linux_aarch64": [], "//tensorflow:linux_ppc64le": [], "//conditions:default": ["@mkl_dnn//:mkldnn_single_threaded"], }), @@ -3222,8 +3224,8 @@ tf_cc_test( name = "eigen_mkldnn_contraction_kernel_test", size = "small", srcs = select({ - "//tensorflow:android_x86": [], - "//tensorflow:arm_any": [], + "//tensorflow:android": [], + "//tensorflow:arm": [], "//tensorflow:ios": [], "//tensorflow:linux_ppc64le": [], ":no_mkldnn_contraction_kernel": [], diff --git a/tensorflow/core/platform/platform.h b/tensorflow/core/platform/platform.h index 3375a6e50eb..a840d7b06e3 100644 --- a/tensorflow/core/platform/platform.h +++ b/tensorflow/core/platform/platform.h @@ -41,22 +41,18 @@ limitations under the License. #elif defined(_WIN32) #define PLATFORM_WINDOWS +#elif defined(__arm__) +#define PLATFORM_POSIX + #elif defined(__EMSCRIPTEN__) #define PLATFORM_PORTABLE_GOOGLE #define PLATFORM_POSIX -// EMSCRIPTEN builds are considered "mobile" for the sake of portability. -#define IS_MOBILE_PLATFORM - -#elif defined(__arm__) || defined(__aarch64__) -// If no platform specified, use: -#define PLATFORM_POSIX // Require an outside macro to tell us if we're building for Raspberry Pi or // another ARM device that's not a mobile platform. -#if !defined(RASPBERRY_PI) && !defined(ARM_NON_MOBILE) && \ - !defined(PLATFORM_GOOGLE) +#if !defined(RASPBERRY_PI) && !defined(ARM_NON_MOBILE) #define IS_MOBILE_PLATFORM -#endif +#endif // !defined(RASPBERRY_PI) && !defined(ARM_NON_MOBILE) #else // If no platform specified, use: