diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-15 16:49:30 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-15 16:53:32 -0800 |
commit | 4bdea2e4023f28ec50035af07c27b14a11c405bc (patch) | |
tree | 71a0100870afdfdcffd24659d35a6e0df89da454 /tensorflow/tensorflow.bzl | |
parent | dcb0666a2be2c78d1f36984ef45910998f19e50b (diff) |
iOS build cleanup.
PiperOrigin-RevId: 179256571
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 425997e41f..611d50bc52 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -103,6 +103,12 @@ def if_ios(a): "//conditions:default": [], }) +def if_ios_x86_64(a): + return select({ + clean_dep("//tensorflow:ios_x86_64"): a, + "//conditions:default": [], + }) + def if_mobile(a): return select({ clean_dep("//tensorflow:android"): a, @@ -177,6 +183,7 @@ def tf_copts(android_optimization_level_override="-O2"): + if_mkl(["-DINTEL_MKL=1", "-DEIGEN_USE_VML", "-fopenmp",]) + if_android_arm(["-mfpu=neon"]) + if_linux_x86_64(["-msse3"]) + + if_ios_x86_64(["-msse4.1"]) + select({ clean_dep("//tensorflow:android"): android_copts, clean_dep("//tensorflow:darwin"): [], |