github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/ngraph/ngraph.BUILD (about)

     1  licenses(["notice"])  # 3-Clause BSD
     2  
     3  exports_files(["LICENSE"])
     4  
     5  cc_library(
     6      name = "ngraph_headers",
     7      hdrs = glob(["src/ngraph/**/*.hpp"]),
     8      visibility = ["//visibility:public"],
     9  )
    10  
    11  cc_library(
    12      name = "ngraph_cpu_backend",
    13      srcs = [
    14          "src/ngraph/runtime/cpu/builder/add.cpp",
    15          "src/ngraph/runtime/cpu/builder/allreduce.cpp",
    16          "src/ngraph/runtime/cpu/builder/argmax.cpp",
    17          "src/ngraph/runtime/cpu/builder/argmin.cpp",
    18          "src/ngraph/runtime/cpu/builder/avg_pool.cpp",
    19          "src/ngraph/runtime/cpu/builder/batch_norm.cpp",
    20          "src/ngraph/runtime/cpu/builder/bounded_relu.cpp",
    21          "src/ngraph/runtime/cpu/builder/broadcast.cpp",
    22          "src/ngraph/runtime/cpu/builder/concat.cpp",
    23          "src/ngraph/runtime/cpu/builder/convert.cpp",
    24          "src/ngraph/runtime/cpu/builder/convert_layout.cpp",
    25          "src/ngraph/runtime/cpu/builder/convolution.cpp",
    26          "src/ngraph/runtime/cpu/builder/dot.cpp",
    27          "src/ngraph/runtime/cpu/builder/function_call.cpp",
    28          "src/ngraph/runtime/cpu/builder/lrn.cpp",
    29          "src/ngraph/runtime/cpu/builder/lstm.cpp",
    30          "src/ngraph/runtime/cpu/builder/matmul_bias.cpp",
    31          "src/ngraph/runtime/cpu/builder/max.cpp",
    32          "src/ngraph/runtime/cpu/builder/max_pool.cpp",
    33          "src/ngraph/runtime/cpu/builder/min.cpp",
    34          "src/ngraph/runtime/cpu/builder/one_hot.cpp",
    35          "src/ngraph/runtime/cpu/builder/pad.cpp",
    36          "src/ngraph/runtime/cpu/builder/product.cpp",
    37          "src/ngraph/runtime/cpu/builder/quantization.cpp",
    38          "src/ngraph/runtime/cpu/builder/quantized_avg_pool.cpp",
    39          "src/ngraph/runtime/cpu/builder/quantized_conv.cpp",
    40          "src/ngraph/runtime/cpu/builder/quantized_max_pool.cpp",
    41          "src/ngraph/runtime/cpu/builder/reduce_function.cpp",
    42          "src/ngraph/runtime/cpu/builder/reduce_function_window.cpp",
    43          "src/ngraph/runtime/cpu/builder/relu.cpp",
    44          "src/ngraph/runtime/cpu/builder/replace_slice.cpp",
    45          "src/ngraph/runtime/cpu/builder/reshape.cpp",
    46          "src/ngraph/runtime/cpu/builder/reverse.cpp",
    47          "src/ngraph/runtime/cpu/builder/reverse_sequence.cpp",
    48          "src/ngraph/runtime/cpu/builder/rnn.cpp",
    49          "src/ngraph/runtime/cpu/builder/select.cpp",
    50          "src/ngraph/runtime/cpu/builder/select_and_scatter.cpp",
    51          "src/ngraph/runtime/cpu/builder/sigmoid.cpp",
    52          "src/ngraph/runtime/cpu/builder/slice.cpp",
    53          "src/ngraph/runtime/cpu/builder/softmax.cpp",
    54          "src/ngraph/runtime/cpu/builder/sum.cpp",
    55          "src/ngraph/runtime/cpu/builder/topk.cpp",
    56          "src/ngraph/runtime/cpu/cpu_backend.cpp",
    57          "src/ngraph/runtime/cpu/cpu_builder.cpp",
    58          "src/ngraph/runtime/cpu/cpu_call_frame.cpp",
    59          "src/ngraph/runtime/cpu/cpu_cse.cpp",
    60          "src/ngraph/runtime/cpu/cpu_executor.cpp",
    61          "src/ngraph/runtime/cpu/cpu_external_function.cpp",
    62          "src/ngraph/runtime/cpu/cpu_kernels.cpp",
    63          "src/ngraph/runtime/cpu/cpu_layout_descriptor.cpp",
    64          "src/ngraph/runtime/cpu/cpu_op_annotations.cpp",
    65          "src/ngraph/runtime/cpu/cpu_tensor_view.cpp",
    66          "src/ngraph/runtime/cpu/cpu_tensor_view_wrapper.cpp",
    67          "src/ngraph/runtime/cpu/cpu_tracing.cpp",
    68          "src/ngraph/runtime/cpu/cpu_visualize_tree.cpp",
    69          "src/ngraph/runtime/cpu/kernel/pad.cpp",
    70          "src/ngraph/runtime/cpu/kernel/reduce_max.cpp",
    71          "src/ngraph/runtime/cpu/kernel/reduce_sum.cpp",
    72          "src/ngraph/runtime/cpu/kernel/reshape.cpp",
    73          "src/ngraph/runtime/cpu/mkldnn_emitter.cpp",
    74          "src/ngraph/runtime/cpu/mkldnn_invoke.cpp",
    75          "src/ngraph/runtime/cpu/mkldnn_utils.cpp",
    76          "src/ngraph/runtime/cpu/op/batch_dot.cpp",
    77          "src/ngraph/runtime/cpu/op/batch_norm_relu.cpp",
    78          "src/ngraph/runtime/cpu/op/bounded_relu.cpp",
    79          "src/ngraph/runtime/cpu/op/conv_add.cpp",
    80          "src/ngraph/runtime/cpu/op/conv_bias.cpp",
    81          "src/ngraph/runtime/cpu/op/conv_relu.cpp",
    82          "src/ngraph/runtime/cpu/op/convert_layout.cpp",
    83          "src/ngraph/runtime/cpu/op/group_conv.cpp",
    84          "src/ngraph/runtime/cpu/op/group_conv_bias.cpp",
    85          "src/ngraph/runtime/cpu/op/halide_op.cpp",
    86          "src/ngraph/runtime/cpu/op/leaky_relu.cpp",
    87          "src/ngraph/runtime/cpu/op/loop_kernel.cpp",
    88          "src/ngraph/runtime/cpu/op/lstm.cpp",
    89          "src/ngraph/runtime/cpu/op/matmul_bias.cpp",
    90          "src/ngraph/runtime/cpu/op/max_pool_with_indices.cpp",
    91          "src/ngraph/runtime/cpu/op/rnn.cpp",
    92          "src/ngraph/runtime/cpu/op/sigmoid_mul.cpp",
    93          "src/ngraph/runtime/cpu/op/update_slice.cpp",
    94          "src/ngraph/runtime/cpu/pass/cpu_assignment.cpp",
    95          "src/ngraph/runtime/cpu/pass/cpu_collapse_dims.cpp",
    96          "src/ngraph/runtime/cpu/pass/cpu_fusion.cpp",
    97          "src/ngraph/runtime/cpu/pass/cpu_horizontal_fusion.cpp",
    98          "src/ngraph/runtime/cpu/pass/cpu_layout.cpp",
    99          "src/ngraph/runtime/cpu/pass/cpu_loop_kernel_fusion.cpp",
   100          "src/ngraph/runtime/cpu/pass/cpu_mat_fusion.cpp",
   101          "src/ngraph/runtime/cpu/pass/cpu_memory_optimization.cpp",
   102          "src/ngraph/runtime/cpu/pass/cpu_post_layout_optimizations.cpp",
   103          "src/ngraph/runtime/cpu/pass/cpu_reshape_sinking.cpp",
   104          "src/ngraph/runtime/cpu/pass/cpu_rnn_fusion.cpp",
   105          "src/ngraph/runtime/cpu/pass/cpu_workspace_insertion.cpp",
   106      ],
   107      hdrs = glob(["src/ngraph/runtime/cpu/**/*.hpp"]) + glob([]),
   108      copts = [
   109          "-I external/ngraph/src",
   110          "-I external/nlohmann_json_lib/include/",
   111          '-D SHARED_LIB_EXT=\\".so\\"',
   112          '-D NGRAPH_VERSION=\\"0.11.0\\"',
   113          "-D NGRAPH_DEX_ONLY",
   114          '-D PROJECT_ROOT_DIR=\\"\\"',
   115      ],
   116      visibility = ["//visibility:public"],
   117      deps = [
   118          ":ngraph_headers",
   119          "@eigen_archive//:eigen",
   120          "@mkl_dnn",
   121          "@nlohmann_json_lib",
   122          "@tbb",
   123      ],
   124      alwayslink = 1,
   125  )
   126  
   127  cc_library(
   128      name = "ngraph_core",
   129      srcs = glob([
   130          "src/ngraph/*.cpp",
   131          "src/ngraph/autodiff/*.cpp",
   132          "src/ngraph/builder/*.cpp",
   133          "src/ngraph/descriptor/*.cpp",
   134          "src/ngraph/descriptor/layout/*.cpp",
   135          "src/ngraph/op/experimental/generate_mask.cpp",
   136          "src/ngraph/op/experimental/quantized_avg_pool.cpp",
   137          "src/ngraph/op/experimental/quantized_conv_bias.cpp",
   138          "src/ngraph/op/experimental/quantized_conv_relu.cpp",
   139          "src/ngraph/op/experimental/quantized_conv.cpp",
   140          "src/ngraph/op/experimental/quantized_max_pool.cpp",
   141          "src/ngraph/op/experimental/shape_of.cpp",
   142          "src/ngraph/op/*.cpp",
   143          "src/ngraph/op/util/*.cpp",
   144          "src/ngraph/pattern/*.cpp",
   145          "src/ngraph/pattern/*.hpp",
   146          "src/ngraph/pass/*.cpp",
   147          "src/ngraph/pass/*.hpp",
   148          "src/ngraph/runtime/*.cpp",
   149          "src/ngraph/type/*.cpp",
   150      ]),
   151      copts = [
   152          "-I external/ngraph/src",
   153          "-I external/nlohmann_json_lib/include/",
   154          '-D SHARED_LIB_EXT=\\".so\\"',
   155          '-D NGRAPH_VERSION=\\"0.11.0\\"',
   156          '-D PROJECT_ROOT_DIR=\\"\\"',
   157      ],
   158      visibility = ["//visibility:public"],
   159      deps = [
   160          ":ngraph_cpu_backend",
   161          ":ngraph_headers",
   162          "@eigen_archive//:eigen",
   163          "@nlohmann_json_lib",
   164      ],
   165      alwayslink = 1,
   166  )