diff options
author | 2016-09-12 16:07:46 -0800 | |
---|---|---|
committer | 2016-09-12 17:16:08 -0700 | |
commit | 56b003d724ecb3e7781165fd4504d058982959c1 (patch) | |
tree | 7a36568edabdc818ef6c63ee503acdb2fdfb6fee /tensorflow/c/BUILD | |
parent | 6de77d7f296c2fce09afa0f86c5237a5d42be153 (diff) |
Add TF_DeleteLibraryHandle, for freeing the memory allocated by TF_LoadLibrary.
This is useful when calling the C API from C++, to avoid memory leaks being
reported.
Change: 132937873
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 50887f70b5..7f86d345d7 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -7,6 +7,7 @@ load( "//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_cuda_library", + "tf_custom_op_library", ) # For platform specific build config @@ -60,6 +61,7 @@ tf_cc_test( name = "c_api_test", size = "small", srcs = ["c_api_test.cc"], + data = [":test_op.so"], linkopts = select({ "//tensorflow:darwin": ["-headerpad_max_install_names"], "//conditions:default": [], @@ -81,6 +83,11 @@ tf_cc_test( ], ) +tf_custom_op_library( + name = "test_op.so", + srcs = ["test_op.cc"], +) + # ----------------------------------------------------------------------------- # Google-internal targets. |