diff options
Diffstat (limited to 'third_party/ngraph/ngraph_tf.BUILD')
| -rw-r--r-- | third_party/ngraph/ngraph_tf.BUILD | 51 |
1 files changed, 27 insertions, 24 deletions
diff --git a/third_party/ngraph/ngraph_tf.BUILD b/third_party/ngraph/ngraph_tf.BUILD index 979318d7c2..dbedca0a03 100644 --- a/third_party/ngraph/ngraph_tf.BUILD +++ b/third_party/ngraph/ngraph_tf.BUILD @@ -10,41 +10,42 @@ load( cc_library( name = "ngraph_tf", srcs = [ - "src/ngraph_api.h", "src/ngraph_api.cc", - "src/ngraph_assign_clusters.h", + "src/ngraph_api.h", "src/ngraph_assign_clusters.cc", - "src/ngraph_builder.h", + "src/ngraph_assign_clusters.h", "src/ngraph_builder.cc", - "src/ngraph_capture_variables.h", + "src/ngraph_builder.h", "src/ngraph_capture_variables.cc", - "src/ngraph_conversions.h", - "src/ngraph_cluster_manager.h", + "src/ngraph_capture_variables.h", "src/ngraph_cluster_manager.cc", - "src/ngraph_deassign_clusters.h", + "src/ngraph_cluster_manager.h", + "src/ngraph_conversions.h", "src/ngraph_deassign_clusters.cc", - "src/ngraph_encapsulate_op.cc", - "src/ngraph_encapsulate_clusters.h", + "src/ngraph_deassign_clusters.h", "src/ngraph_encapsulate_clusters.cc", - "src/ngraph_freshness_tracker.h", + "src/ngraph_encapsulate_clusters.h", + "src/ngraph_encapsulate_op.cc", "src/ngraph_freshness_tracker.cc", - "src/ngraph_mark_for_clustering.h", + "src/ngraph_freshness_tracker.h", "src/ngraph_mark_for_clustering.cc", - "src/ngraph_rewrite_pass.cc", - "src/ngraph_rewrite_for_tracking.h", + "src/ngraph_mark_for_clustering.h", "src/ngraph_rewrite_for_tracking.cc", + "src/ngraph_rewrite_for_tracking.h", + "src/ngraph_rewrite_pass.cc", "src/ngraph_tracked_variable.cc", - "src/ngraph_utils.h", "src/ngraph_utils.cc", + "src/ngraph_utils.h", + "src/ngraph_version_utils.h", + "src/tf_deadness_analysis.cc", + "src/tf_deadness_analysis.h", "src/tf_graphcycles.cc", + "src/tf_graphcycles.h", "logging/ngraph_log.h", "logging/ngraph_log.cc", "logging/tf_graph_writer.h", "logging/tf_graph_writer.cc", ], - hdrs = [ - "src/tf_graphcycles.h", - ], deps = [ "@org_tensorflow//tensorflow/core:protos_all_proto_text", "@org_tensorflow//tensorflow/core:framework_headers_lib", @@ -64,17 +65,19 @@ tf_cc_test( name = "ngraph_tf_tests", size = "small", srcs = [ - "test/tf_exec.cpp", "test/conversions.cpp", - "test/padding.cpp", "test/graph_rewrites/assign_clusters.cc", - "test/test_utilities.h", - "test/test_utilities.cpp", + "test/graph_rewrites/deadness_test.cc", + "test/main.cpp", + "test/opexecuter.cpp", + "test/opexecuter.h", + "test/padding.cpp", + "test/test_array_ops.cpp", "test/test_math_ops.cpp", "test/test_nn_ops.cpp", - "test/opexecuter.h", - "test/opexecuter.cpp", - "test/main.cpp", + "test/test_utilities.cpp", + "test/test_utilities.h", + "test/tf_exec.cpp", ], deps = [ ":ngraph_tf", |