diff options
| author |  Avijit <30507445+avijit-nervana@users.noreply.github.com> | 2018-07-24 23:35:27 -0700 |
|---|---|---|
| committer |  GitHub <noreply@github.com> | 2018-07-24 23:35:27 -0700 |
| commit | 121e0161c5a7273c5a59f1e10a8577428c685796 (patch) | |
| tree | fb6b99b4af3accd9c68d05442be95f9250b59604 /tensorflow/tensorflow.bzl | |
| parent | 80fb8679ab14ba3d180e8eb22da11509a15b9219 (diff) | |
nGraph integration with TensorFlow
* Added nGraph bridge as a third_party to be built with TensorFlow based on user selection.
* Added a limited set of C++ unit tests to verify the correctness of the computation
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 e4241667ad..5884870daa 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -24,6 +24,10 @@ load( "if_mkl", "if_mkl_lnx_x64" ) +load( + "//third_party/ngraph:build_defs.bzl", + "if_ngraph", +) def register_extension_info(**kwargs): pass @@ -214,6 +218,7 @@ def tf_copts(android_optimization_level_override="-O2", is_external=False): + if_cuda(["-DGOOGLE_CUDA=1"]) + if_tensorrt(["-DGOOGLE_TENSORRT=1"]) + if_mkl(["-DINTEL_MKL=1", "-DEIGEN_USE_VML"]) + + if_ngraph(["-DINTEL_NGRAPH=1"]) + if_mkl_lnx_x64(["-fopenmp"]) + if_android_arm(["-mfpu=neon"]) + if_linux_x86_64(["-msse3"]) |