diff options
Diffstat (limited to 'tensorflow/BUILD')
-rw-r--r-- | tensorflow/BUILD | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 388ca3f293..cee7114d3e 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( |