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  )