43 lines
903 B
Python
43 lines
903 B
Python
load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
|
|
|
|
package(
|
|
default_visibility = [
|
|
"//tensorflow:__subpackages__",
|
|
],
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
exports_files(["LICENSE"])
|
|
|
|
config_setting(
|
|
name = "build_with_mkl_dnn_only",
|
|
define_values = {
|
|
"build_with_mkl": "true",
|
|
"build_with_mkl_dnn_only": "true",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
config_setting(
|
|
name = "build_with_mkl_dnn_v1_only",
|
|
define_values = {
|
|
"build_with_mkl": "true",
|
|
"build_with_mkl_dnn_v1_only": "true",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
config_setting(
|
|
name = "build_with_mkldnn_threadpool",
|
|
define_values = {
|
|
"build_with_mkl": "true",
|
|
"build_with_mkldnn_threadpool": "true",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
bzl_library(
|
|
name = "build_defs_bzl",
|
|
srcs = ["build_defs.bzl"],
|
|
)
|