diff options
| author |  Jonathan Hseu <jhseu@google.com> | 2016-09-29 15:05:32 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-29 16:17:09 -0700 |
| commit | 1283b84a49a9f5e14aca833cf981b61848aaf916 (patch) | |
| tree | ffba9d2d8ba549bd5981cc84748d2db8858fc676 /tensorflow/contrib/framework/BUILD | |
| parent | ef9f5fee0a079f6bed445064e8e9d18fb7a904d8 (diff) | |
Merge changes from github.
Change: 134721831
Diffstat (limited to 'tensorflow/contrib/framework/BUILD')
| -rw-r--r-- | tensorflow/contrib/framework/BUILD | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/tensorflow/contrib/framework/BUILD b/tensorflow/contrib/framework/BUILD index db8d10748e..ca625eccb3 100644 --- a/tensorflow/contrib/framework/BUILD +++ b/tensorflow/contrib/framework/BUILD @@ -8,6 +8,9 @@ exports_files(["LICENSE"]) package(default_visibility = ["//tensorflow:__subpackages__"]) load("//tensorflow:tensorflow.bzl", "cuda_py_test") +load("//tensorflow:tensorflow.bzl", "tf_custom_op_library") +load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_py") +load("//tensorflow:tensorflow.bzl", "tf_gen_op_libs") py_library( name = "framework_py", @@ -23,7 +26,41 @@ py_library( "python/ops/prettyprint_ops.py", "python/ops/variables.py", ], + data = [ + ":python/ops/_variable_ops.so", + ], srcs_version = "PY2AND3", + deps = [ + ":gen_variable_ops", + ], +) + +tf_custom_op_library( + name = "python/ops/_variable_ops.so", + srcs = [ + "kernels/zero_initializer_op.cc", + "kernels/zero_initializer_op.h", + "ops/variable_ops.cc", + ], +) + +tf_gen_op_libs( + op_lib_names = ["variable_ops"], +) + +cc_library( + name = "all_ops", + deps = [ + ":variable_ops_op_lib", + ], +) + +tf_gen_op_wrapper_py( + name = "gen_variable_ops", + out = "python/ops/gen_variable_ops.py", + deps = [ + ":variable_ops_op_lib", + ], ) py_test( |