github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/ngraph/ngraph_tf.BUILD (about) 1 licenses(["notice"]) # 3-Clause BSD 2 3 exports_files(["LICENSE"]) 4 5 load( 6 "@org_tensorflow//tensorflow:tensorflow.bzl", 7 "tf_cc_test", 8 ) 9 10 cc_library( 11 name = "ngraph_tf", 12 srcs = [ 13 "logging/ngraph_log.cc", 14 "logging/ngraph_log.h", 15 "logging/tf_graph_writer.cc", 16 "logging/tf_graph_writer.h", 17 "src/ngraph_api.cc", 18 "src/ngraph_api.h", 19 "src/ngraph_assign_clusters.cc", 20 "src/ngraph_assign_clusters.h", 21 "src/ngraph_backend_manager.cc", 22 "src/ngraph_backend_manager.h", 23 "src/ngraph_builder.cc", 24 "src/ngraph_builder.h", 25 "src/ngraph_capture_variables.cc", 26 "src/ngraph_capture_variables.h", 27 "src/ngraph_cluster_manager.cc", 28 "src/ngraph_cluster_manager.h", 29 "src/ngraph_conversions.h", 30 "src/ngraph_deassign_clusters.cc", 31 "src/ngraph_deassign_clusters.h", 32 "src/ngraph_encapsulate_clusters.cc", 33 "src/ngraph_encapsulate_clusters.h", 34 "src/ngraph_encapsulate_op.cc", 35 "src/ngraph_freshness_tracker.cc", 36 "src/ngraph_freshness_tracker.h", 37 "src/ngraph_mark_for_clustering.cc", 38 "src/ngraph_mark_for_clustering.h", 39 "src/ngraph_rewrite_for_tracking.cc", 40 "src/ngraph_rewrite_for_tracking.h", 41 "src/ngraph_rewrite_pass.cc", 42 "src/ngraph_tracked_variable.cc", 43 "src/ngraph_utils.cc", 44 "src/ngraph_utils.h", 45 "src/ngraph_version_utils.h", 46 "src/tf_deadness_analysis.cc", 47 "src/tf_deadness_analysis.h", 48 "src/tf_graphcycles.cc", 49 "src/tf_graphcycles.h", 50 ], 51 copts = [ 52 "-I external/ngraph_tf/src", 53 "-I external/ngraph_tf/logging", 54 "-I external/ngraph/src", 55 ], 56 visibility = ["//visibility:public"], 57 deps = [ 58 "@com_google_absl//absl/container:container_memory", 59 "@com_google_absl//absl/container:flat_hash_set", 60 "@com_google_absl//absl/types:variant", 61 "@ngraph//:ngraph_core", 62 "@org_tensorflow//tensorflow/core:core_cpu_headers_lib", 63 "@org_tensorflow//tensorflow/core:framework_headers_lib", 64 "@org_tensorflow//tensorflow/core:protos_all_proto_text", 65 ], 66 alwayslink = 1, 67 ) 68 69 tf_cc_test( 70 name = "ngraph_tf_tests", 71 size = "small", 72 srcs = [ 73 "test/conversions.cpp", 74 "test/graph_rewrites/assign_clusters.cc", 75 "test/graph_rewrites/deadness_test.cc", 76 "test/main.cpp", 77 "test/opexecuter.cpp", 78 "test/opexecuter.h", 79 "test/padding.cpp", 80 "test/test_array_ops.cpp", 81 "test/test_math_ops.cpp", 82 "test/test_nn_ops.cpp", 83 "test/test_utilities.cpp", 84 "test/test_utilities.h", 85 "test/tf_exec.cpp", 86 ], 87 extra_copts = [ 88 "-fexceptions ", 89 "-I external/ngraph_tf/src", 90 "-I external/ngraph_tf/logging", 91 "-I external/ngraph/src", 92 ], 93 deps = [ 94 ":ngraph_tf", 95 "@com_google_googletest//:gtest", 96 "@org_tensorflow//tensorflow/cc:cc_ops", 97 "@org_tensorflow//tensorflow/cc:client_session", 98 "@org_tensorflow//tensorflow/core:tensorflow", 99 ], 100 )