aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/BUILD')
-rw-r--r--tensorflow/BUILD12
1 files changed, 11 insertions, 1 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD
index 60db234c9c..529707b6fc 100644
--- a/tensorflow/BUILD
+++ b/tensorflow/BUILD
@@ -416,6 +416,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 = [
@@ -432,6 +440,8 @@ load(
"if_mkl",
)
+load("//third_party/ngraph:build_defs.bzl", "if_ngraph")
+
filegroup(
name = "intel_binary_blob",
data = if_mkl(
@@ -576,7 +586,7 @@ tf_cc_shared_object(
"//tensorflow/cc:scope",
"//tensorflow/cc/profiler",
"//tensorflow/core:tensorflow",
- ],
+ ] + if_ngraph(["@ngraph_tf//:ngraph_tf"])
)
exports_files(