| licenses(["notice"]) # 3-Clause BSD |
| |
| exports_files(["LICENSE"]) |
| |
| cc_library( |
| name = "ngraph_headers", |
| hdrs = glob(["src/ngraph/**/*.hpp"]), |
| visibility = ["//visibility:public"], |
| ) |
| |
| cc_library( |
| name = "ngraph_cpu_backend", |
| srcs = [ |
| "src/ngraph/runtime/cpu/builder/add.cpp", |
| "src/ngraph/runtime/cpu/builder/allreduce.cpp", |
| "src/ngraph/runtime/cpu/builder/argmax.cpp", |
| "src/ngraph/runtime/cpu/builder/argmin.cpp", |
| "src/ngraph/runtime/cpu/builder/avg_pool.cpp", |
| "src/ngraph/runtime/cpu/builder/batch_norm.cpp", |
| "src/ngraph/runtime/cpu/builder/bounded_relu.cpp", |
| "src/ngraph/runtime/cpu/builder/broadcast.cpp", |
| "src/ngraph/runtime/cpu/builder/concat.cpp", |
| "src/ngraph/runtime/cpu/builder/convert.cpp", |
| "src/ngraph/runtime/cpu/builder/convert_layout.cpp", |
| "src/ngraph/runtime/cpu/builder/convolution.cpp", |
| "src/ngraph/runtime/cpu/builder/dot.cpp", |
| "src/ngraph/runtime/cpu/builder/function_call.cpp", |
| "src/ngraph/runtime/cpu/builder/lrn.cpp", |
| "src/ngraph/runtime/cpu/builder/lstm.cpp", |
| "src/ngraph/runtime/cpu/builder/matmul_bias.cpp", |
| "src/ngraph/runtime/cpu/builder/max.cpp", |
| "src/ngraph/runtime/cpu/builder/max_pool.cpp", |
| "src/ngraph/runtime/cpu/builder/min.cpp", |
| "src/ngraph/runtime/cpu/builder/one_hot.cpp", |
| "src/ngraph/runtime/cpu/builder/pad.cpp", |
| "src/ngraph/runtime/cpu/builder/product.cpp", |
| "src/ngraph/runtime/cpu/builder/quantization.cpp", |
| "src/ngraph/runtime/cpu/builder/quantized_avg_pool.cpp", |
| "src/ngraph/runtime/cpu/builder/quantized_conv.cpp", |
| "src/ngraph/runtime/cpu/builder/quantized_max_pool.cpp", |
| "src/ngraph/runtime/cpu/builder/reduce_function.cpp", |
| "src/ngraph/runtime/cpu/builder/reduce_function_window.cpp", |
| "src/ngraph/runtime/cpu/builder/relu.cpp", |
| "src/ngraph/runtime/cpu/builder/replace_slice.cpp", |
| "src/ngraph/runtime/cpu/builder/reshape.cpp", |
| "src/ngraph/runtime/cpu/builder/reverse.cpp", |
| "src/ngraph/runtime/cpu/builder/reverse_sequence.cpp", |
| "src/ngraph/runtime/cpu/builder/rnn.cpp", |
| "src/ngraph/runtime/cpu/builder/select.cpp", |
| "src/ngraph/runtime/cpu/builder/select_and_scatter.cpp", |
| "src/ngraph/runtime/cpu/builder/sigmoid.cpp", |
| "src/ngraph/runtime/cpu/builder/slice.cpp", |
| "src/ngraph/runtime/cpu/builder/softmax.cpp", |
| "src/ngraph/runtime/cpu/builder/sum.cpp", |
| "src/ngraph/runtime/cpu/builder/topk.cpp", |
| "src/ngraph/runtime/cpu/cpu_backend.cpp", |
| "src/ngraph/runtime/cpu/cpu_builder.cpp", |
| "src/ngraph/runtime/cpu/cpu_call_frame.cpp", |
| "src/ngraph/runtime/cpu/cpu_cse.cpp", |
| "src/ngraph/runtime/cpu/cpu_executor.cpp", |
| "src/ngraph/runtime/cpu/cpu_external_function.cpp", |
| "src/ngraph/runtime/cpu/cpu_kernels.cpp", |
| "src/ngraph/runtime/cpu/cpu_layout_descriptor.cpp", |
| "src/ngraph/runtime/cpu/cpu_op_annotations.cpp", |
| "src/ngraph/runtime/cpu/cpu_tensor_view.cpp", |
| "src/ngraph/runtime/cpu/cpu_tensor_view_wrapper.cpp", |
| "src/ngraph/runtime/cpu/cpu_tracing.cpp", |
| "src/ngraph/runtime/cpu/cpu_visualize_tree.cpp", |
| "src/ngraph/runtime/cpu/kernel/pad.cpp", |
| "src/ngraph/runtime/cpu/kernel/reduce_max.cpp", |
| "src/ngraph/runtime/cpu/kernel/reduce_sum.cpp", |
| "src/ngraph/runtime/cpu/kernel/reshape.cpp", |
| "src/ngraph/runtime/cpu/mkldnn_emitter.cpp", |
| "src/ngraph/runtime/cpu/mkldnn_invoke.cpp", |
| "src/ngraph/runtime/cpu/mkldnn_utils.cpp", |
| "src/ngraph/runtime/cpu/op/batch_dot.cpp", |
| "src/ngraph/runtime/cpu/op/batch_norm_relu.cpp", |
| "src/ngraph/runtime/cpu/op/bounded_relu.cpp", |
| "src/ngraph/runtime/cpu/op/conv_add.cpp", |
| "src/ngraph/runtime/cpu/op/conv_bias.cpp", |
| "src/ngraph/runtime/cpu/op/conv_relu.cpp", |
| "src/ngraph/runtime/cpu/op/convert_layout.cpp", |
| "src/ngraph/runtime/cpu/op/group_conv.cpp", |
| "src/ngraph/runtime/cpu/op/group_conv_bias.cpp", |
| "src/ngraph/runtime/cpu/op/halide_op.cpp", |
| "src/ngraph/runtime/cpu/op/leaky_relu.cpp", |
| "src/ngraph/runtime/cpu/op/loop_kernel.cpp", |
| "src/ngraph/runtime/cpu/op/lstm.cpp", |
| "src/ngraph/runtime/cpu/op/matmul_bias.cpp", |
| "src/ngraph/runtime/cpu/op/max_pool_with_indices.cpp", |
| "src/ngraph/runtime/cpu/op/rnn.cpp", |
| "src/ngraph/runtime/cpu/op/sigmoid_mul.cpp", |
| "src/ngraph/runtime/cpu/op/update_slice.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_assignment.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_collapse_dims.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_fusion.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_horizontal_fusion.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_layout.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_loop_kernel_fusion.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_mat_fusion.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_memory_optimization.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_post_layout_optimizations.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_reshape_sinking.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_rnn_fusion.cpp", |
| "src/ngraph/runtime/cpu/pass/cpu_workspace_insertion.cpp", |
| ], |
| hdrs = glob(["src/ngraph/runtime/cpu/**/*.hpp"]) + glob([]), |
| copts = [ |
| "-I external/ngraph/src", |
| "-I external/nlohmann_json_lib/include/", |
| "-D SHARED_LIB_EXT=\".so\"", |
| "-D NGRAPH_VERSION=\"0.11.0\"", |
| "-D NGRAPH_DEX_ONLY", |
| "-D PROJECT_ROOT_DIR=\"\"", |
| ], |
| visibility = ["//visibility:public"], |
| deps = [ |
| ":ngraph_headers", |
| "@eigen_archive//:eigen", |
| "@mkl_dnn_v1//:mkl_dnn", |
| "@nlohmann_json_lib", |
| "@tbb", |
| ], |
| alwayslink = 1, |
| ) |
| |
| cc_library( |
| name = "ngraph_core", |
| srcs = glob([ |
| "src/ngraph/*.cpp", |
| "src/ngraph/autodiff/*.cpp", |
| "src/ngraph/builder/*.cpp", |
| "src/ngraph/descriptor/*.cpp", |
| "src/ngraph/descriptor/layout/*.cpp", |
| "src/ngraph/op/experimental/generate_mask.cpp", |
| "src/ngraph/op/experimental/quantized_avg_pool.cpp", |
| "src/ngraph/op/experimental/quantized_conv_bias.cpp", |
| "src/ngraph/op/experimental/quantized_conv_relu.cpp", |
| "src/ngraph/op/experimental/quantized_conv.cpp", |
| "src/ngraph/op/experimental/quantized_max_pool.cpp", |
| "src/ngraph/op/experimental/shape_of.cpp", |
| "src/ngraph/op/*.cpp", |
| "src/ngraph/op/util/*.cpp", |
| "src/ngraph/pattern/*.cpp", |
| "src/ngraph/pattern/*.hpp", |
| "src/ngraph/pass/*.cpp", |
| "src/ngraph/pass/*.hpp", |
| "src/ngraph/runtime/*.cpp", |
| "src/ngraph/type/*.cpp", |
| ]), |
| copts = [ |
| "-I external/ngraph/src", |
| "-I external/nlohmann_json_lib/include/", |
| "-D SHARED_LIB_EXT=\\\".so\\\"", |
| "-D NGRAPH_VERSION=\\\"0.11.0\\\"", |
| "-D PROJECT_ROOT_DIR=\\\"\\\"", |
| ], |
| visibility = ["//visibility:public"], |
| deps = [ |
| ":ngraph_cpu_backend", |
| ":ngraph_headers", |
| "@eigen_archive//:eigen", |
| "@nlohmann_json_lib", |
| ], |
| alwayslink = 1, |
| ) |