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/BUILD | |
| 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/BUILD')
| -rw-r--r-- | tensorflow/BUILD | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 51eea94847..6d443eb9f2 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -24,6 +24,8 @@ load( "gen_api_init_files", # @unused ) +load("//third_party/ngraph:build_defs.bzl", "if_ngraph") + # Config setting for determining if we are building for Android. config_setting( name = "android", @@ -408,6 +410,14 @@ config_setting( visibility = ["//visibility:public"], ) +# This flag is set from the configure step when the user selects with nGraph option. +# By default it should be false +config_setting( + name = "with_ngraph_support", + values = {"define": "with_ngraph_support=true"}, + visibility = ["//visibility:public"], +) + package_group( name = "internal", packages = [ @@ -540,7 +550,7 @@ tf_cc_shared_object( "//tensorflow/c:version_script.lds", "//tensorflow/c/eager:c_api", "//tensorflow/core:tensorflow", - ], + ] ) tf_cc_shared_object( @@ -568,7 +578,7 @@ tf_cc_shared_object( "//tensorflow/cc:scope", "//tensorflow/cc/profiler", "//tensorflow/core:tensorflow", - ], + ] + if_ngraph(["@ngraph_tf//:ngraph_tf"]) ) exports_files( |