diff options
| author |  Rohan Jain <rohanj@google.com> | 2017-03-23 16:36:51 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-23 17:52:19 -0700 |
| commit | b2ac11c0797447eb36d427a0907fb2f05af57fcc (patch) | |
| tree | eafeef30eed4451bc55c421d966b01d2fda96327 /tensorflow/contrib/factorization/BUILD | |
| parent | 494fe43926f8928bb3b10c1729abea9be5113f21 (diff) | |
Internal changes
Change: 151077399
Diffstat (limited to 'tensorflow/contrib/factorization/BUILD')
| -rw-r--r-- | tensorflow/contrib/factorization/BUILD | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/tensorflow/contrib/factorization/BUILD b/tensorflow/contrib/factorization/BUILD index b329bb9595..338e67e953 100644 --- a/tensorflow/contrib/factorization/BUILD +++ b/tensorflow/contrib/factorization/BUILD @@ -13,8 +13,9 @@ load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_py") load("//tensorflow:tensorflow.bzl", "tf_gen_op_libs") load("//tensorflow:tensorflow.bzl", "tf_py_test") load("//tensorflow:tensorflow.bzl", "cuda_py_test") +load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library") -py_library( +tf_custom_op_py_library( name = "factorization_py", srcs = [ "__init__.py", @@ -23,10 +24,14 @@ py_library( "python/ops/gmm.py", "python/ops/gmm_ops.py", ], - data = [ + dso = [ ":python/ops/_clustering_ops.so", ":python/ops/_factorization_ops.so", ], + kernels = [ + ":all_ops", + "//tensorflow/contrib/factorization/kernels:all_kernels", + ], srcs_version = "PY2AND3", deps = [ ":gen_clustering_ops", |