xref: /aosp_15_r20/external/tensorflow/third_party/mkl_dnn/BUILD (revision b6fb3261f9314811a0f4371741dbb8839866f948)
1load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
2
3package(
4    default_visibility = [
5        "//tensorflow:__subpackages__",
6    ],
7    licenses = ["notice"],
8)
9
10exports_files(["LICENSE"])
11
12config_setting(
13    name = "build_with_mkl_opensource",
14    define_values = {
15        "build_with_mkl": "true",
16        "build_with_mkl_opensource": "true",
17    },
18    visibility = ["//visibility:public"],
19)
20
21config_setting(
22    name = "build_with_mkldnn_openmp",
23    define_values = {
24        "build_with_mkl": "true",
25        "build_with_openmp": "true",
26    },
27    visibility = ["//visibility:public"],
28)
29
30config_setting(
31    name = "build_with_mkl_aarch64_openmp",
32    define_values = {
33        "build_with_mkl_aarch64": "true",
34        "build_with_openmp": "true",
35    },
36    visibility = ["//visibility:public"],
37)
38
39config_setting(
40    name = "build_with_mkl_aarch64",
41    define_values = {
42        "build_with_mkl_aarch64": "true",
43    },
44    visibility = ["//visibility:public"],
45)
46
47bzl_library(
48    name = "build_defs_bzl",
49    srcs = ["build_defs.bzl"],
50)
51