diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 11:02:27 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 11:02:41 -0700 |
commit | c72c36145b01ba71b88731843fc36914f721a0e8 (patch) | |
tree | e1f6fd4048ff5dc7c0b3330cf80a17283e57c42a /tensorflow/tensorflow.bzl | |
parent | a2119c81894e99160978a444f2e8d9431d0f7abb (diff) | |
parent | bc6be507c71046dfc889a90e3949a903d5d1e6eb (diff) |
Merge pull request #21122 from NervanaSystems:master
PiperOrigin-RevId: 209623532
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index c7766f384e..6d6e8941c5 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -31,6 +31,10 @@ load( "//third_party/mkl_dnn:build_defs.bzl", "if_mkl_open_source_only", ) +load( + "//third_party/ngraph:build_defs.bzl", + "if_ngraph", +) def register_extension_info(**kwargs): pass @@ -233,6 +237,7 @@ def tf_copts(android_optimization_level_override = "-O2", is_external = False): if_tensorrt(["-DGOOGLE_TENSORRT=1"]) + if_mkl(["-DINTEL_MKL=1", "-DEIGEN_USE_VML"]) + if_mkl_open_source_only(["-DINTEL_MKL_DNN_ONLY"]) + + if_ngraph(["-DINTEL_NGRAPH=1"]) + if_mkl_lnx_x64(["-fopenmp"]) + if_android_arm(["-mfpu=neon"]) + if_linux_x86_64(["-msse3"]) + |