diff options
author | 2018-02-16 22:05:07 -0800 | |
---|---|---|
committer | 2018-02-16 22:09:20 -0800 | |
commit | fa8c4d16288e3bee4a014b4d51d22dd361721ff4 (patch) | |
tree | f42d8b707cc5c7d9cb478fb36065d1b22fe95253 /tensorflow/c/BUILD | |
parent | 02bbb131b78fb0924675809ed5b549e594a51ac1 (diff) |
Added an experimental C API TF_EnableXLACompilation() to enable XLA compilation.
Also ran "buildozer warn //third_party/tensorflow/c/BUILD" and removed an unused symbol.
PiperOrigin-RevId: 186081948
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 26 |
1 files changed, 19 insertions, 7 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 9060c58c13..85cfa98908 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -12,12 +12,6 @@ load( "tf_custom_op_library", ) -# For platform specific build config -load( - "//tensorflow/core:platform/default/build_config.bzl", - "tf_kernel_tests_linkstatic", -) - # ----------------------------------------------------------------------------- # Public targets @@ -101,6 +95,24 @@ tf_cuda_library( }), ) +tf_cuda_library( + name = "c_api_experimental", + srcs = [ + "c_api_experimental.cc", + ], + hdrs = [ + "c_api_experimental.h", + ], + copts = tf_copts(), + visibility = ["//visibility:public"], + deps = [ + ":c_api", + ":c_api_internal", + "//tensorflow/compiler/jit/legacy_flags:mark_for_compilation_pass_flags", + "//tensorflow/core:protos_all_cc", + ], +) + exports_files( [ "version_script.lds", @@ -148,7 +160,7 @@ tf_cuda_library( ], deps = [ ":c_api", - "//tensorflow/compiler/jit/legacy_flags:mark_for_compilation_pass_flags", + ":c_api_experimental", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "//tensorflow/core:session_options", |