diff options
Diffstat (limited to 'tensorflow/python/BUILD')
-rw-r--r-- | tensorflow/python/BUILD | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 07c74ec71d..e6d78301a5 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -44,6 +44,10 @@ load("//tensorflow/core:platform/default/build_config_root.bzl", "tf_additional_ load("//tensorflow/core:platform/default/build_config_root.bzl", "tf_additional_mpi_deps") load("//tensorflow/core:platform/default/build_config_root.bzl", "tf_additional_gdr_deps") load("//tensorflow/core:platform/default/build_config_root.bzl", "if_static") +load( + "//third_party/ngraph:build_defs.bzl", + "if_ngraph", +) py_library( name = "python", @@ -3816,7 +3820,9 @@ tf_py_wrap_cc( tf_additional_plugin_deps() + tf_additional_verbs_deps() + tf_additional_mpi_deps() + - tf_additional_gdr_deps()), + tf_additional_gdr_deps()) + if_ngraph([ + "@ngraph_tf//:ngraph_tf", + ]), ) # ** Targets for Windows build (start) ** |