diff options
Diffstat (limited to 'third_party/ngraph')
| -rw-r--r-- | third_party/ngraph/build_defs.bzl | 10 | ||||
| -rw-r--r-- | third_party/ngraph/ngraph_tf.BUILD | 3 |
2 files changed, 5 insertions, 8 deletions
diff --git a/third_party/ngraph/build_defs.bzl b/third_party/ngraph/build_defs.bzl index 2c9027a6b8..8ad7515aed 100644 --- a/third_party/ngraph/build_defs.bzl +++ b/third_party/ngraph/build_defs.bzl @@ -1,16 +1,14 @@ def clean_dep(dep): return str(Label(dep)) -def if_ngraph(a): +def if_ngraph(if_true, if_false = []): """Shorthand for select()'ing on whether we're building with nGraph support. Returns a select statement which evaluates to if_true if we're building with nGraph. Otherwise, the select statement evaluates to default. """ - ret_val = select({ - clean_dep("//tensorflow:with_ngraph_support"): a, - "//conditions:default": [] + return select({ + clean_dep("//tensorflow:with_ngraph_support"): if_true, + "//conditions:default": if_false }) - - return ret_val diff --git a/third_party/ngraph/ngraph_tf.BUILD b/third_party/ngraph/ngraph_tf.BUILD index bbac74db0f..838f022222 100644 --- a/third_party/ngraph/ngraph_tf.BUILD +++ b/third_party/ngraph/ngraph_tf.BUILD @@ -26,8 +26,7 @@ cc_library( cc_library( name = "ngraph_tf", - srcs = - [ + srcs = [ "src/ngraph_builder.h", "src/ngraph_builder.cc", "src/ngraph_cluster.h", |