diff options
Diffstat (limited to 'tensorflow/BUILD')
-rw-r--r-- | tensorflow/BUILD | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD index a2e74f40c3..1956cb0763 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -110,6 +110,34 @@ config_setting( visibility = ["//visibility:public"], ) +# TODO(jhseu): Enable on other platforms other than Linux. +config_setting( + name = "with_jemalloc", + values = { + "cpu": "k8", + "define": "with_jemalloc=true", + }, + visibility = ["//visibility:public"], +) + +config_setting( + name = "with_gcp_support", + values = {"define": "with_gcp_support=true"}, + visibility = ["//visibility:public"], +) + +config_setting( + name = "with_hdfs_support", + values = {"define": "with_hdfs_support=true"}, + visibility = ["//visibility:public"], +) + +config_setting( + name = "with_xla_support", + values = {"define": "with_xla_support=true"}, + visibility = ["//visibility:public"], +) + package_group( name = "internal", packages = ["//tensorflow/..."], @@ -321,6 +349,8 @@ cc_binary( deps = [ "//tensorflow/c:c_api", "//tensorflow/cc:cc_ops", + "//tensorflow/cc:client_session", + "//tensorflow/cc:scope", "//tensorflow/core:tensorflow", ], ) |