diff options
| author |  avijit-nervana <avijit.chakraborty@intel.com> | 2018-07-31 23:04:56 -0700 |
|---|---|---|
| committer | avijit-nervana <avijit.chakraborty@intel.com> | 2018-07-31 23:04:56 -0700 |
| commit | 2f8b328d2571625161b3c0da0a9b25b907a0927e (patch) | |
| tree | d264a0d5dbfcd774cccd0f2c23d7616897c1b894 /third_party/ngraph | |
| parent | 2f3e97cf0ee50ee0e55ab1a3795cc82537426e8c (diff) | |
| parent | 7ca6ee15555db77c09861fc7e84e5181001da07d (diff) | |
Merge remote-tracking branch 'upstream/master' and changes
based on PR review comments.
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", |