aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/ngraph
diff options
context:
space:
mode:
authorGravatar avijit-nervana <avijit.chakraborty@intel.com>2018-09-05 22:56:20 -0700
committerGravatar avijit-nervana <avijit.chakraborty@intel.com>2018-09-05 22:56:20 -0700
commit18b80bbd4b8db8bd35afad7264258c1c5c269226 (patch)
tree026d004f1a72ba09896f01a924212fcdaef9494e /third_party/ngraph
parentdf7fe4b73609a5ba2a5aa25a77882eb931cd0279 (diff)
Updated with more unit tests
Diffstat (limited to 'third_party/ngraph')
-rw-r--r--third_party/ngraph/ngraph.BUILD4
-rw-r--r--third_party/ngraph/ngraph_tf.BUILD4
2 files changed, 6 insertions, 2 deletions
diff --git a/third_party/ngraph/ngraph.BUILD b/third_party/ngraph/ngraph.BUILD
index 426d49c542..1fd1b8e8e0 100644
--- a/third_party/ngraph/ngraph.BUILD
+++ b/third_party/ngraph/ngraph.BUILD
@@ -101,7 +101,7 @@ cc_library(
"-I external/ngraph/src",
"-I external/nlohmann_json_lib/include/",
'-D SHARED_LIB_EXT=\\".so\\"',
- '-D NGRAPH_VERSION=\\"0.5.0\\"',
+ '-D NGRAPH_VERSION=\\"0.7.0\\"',
"-D NGRAPH_DEX_ONLY",
],
visibility = ["//visibility:public"],
@@ -135,7 +135,7 @@ cc_library(
"-I external/ngraph/src",
"-I external/nlohmann_json_lib/include/",
'-D SHARED_LIB_EXT=\\".so\\"',
- '-D NGRAPH_VERSION=\\"0.5.0\\"',
+ '-D NGRAPH_VERSION=\\"0.7.0\\"',
],
visibility = ["//visibility:public"],
alwayslink = 1,
diff --git a/third_party/ngraph/ngraph_tf.BUILD b/third_party/ngraph/ngraph_tf.BUILD
index 7577a4014d..979318d7c2 100644
--- a/third_party/ngraph/ngraph_tf.BUILD
+++ b/third_party/ngraph/ngraph_tf.BUILD
@@ -70,6 +70,10 @@ tf_cc_test(
"test/graph_rewrites/assign_clusters.cc",
"test/test_utilities.h",
"test/test_utilities.cpp",
+ "test/test_math_ops.cpp",
+ "test/test_nn_ops.cpp",
+ "test/opexecuter.h",
+ "test/opexecuter.cpp",
"test/main.cpp",
],
deps = [