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/BUILD | |
| parent | a2119c81894e99160978a444f2e8d9431d0f7abb (diff) | |
| parent | bc6be507c71046dfc889a90e3949a903d5d1e6eb (diff) | |
Merge pull request #21122 from NervanaSystems:master
PiperOrigin-RevId: 209623532
Diffstat (limited to 'tensorflow/BUILD')
| -rw-r--r-- | tensorflow/BUILD | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 94e059b914..9cc4c4567b 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -23,6 +23,10 @@ load( "//tensorflow/python/tools/api/generator:api_gen.bzl", "gen_api_init_files", # @unused ) +load( + "//third_party/ngraph:build_defs.bzl", + "if_ngraph", +) # Config setting used when building for products # which requires restricted licenses to be avoided. @@ -411,6 +415,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 = [ @@ -563,7 +575,7 @@ tf_cc_shared_object( "//tensorflow/cc:scope", "//tensorflow/cc/profiler", "//tensorflow/core:tensorflow", - ], + ] + if_ngraph(["@ngraph_tf//:ngraph_tf"]), ) exports_files( |