diff options
Diffstat (limited to 'tensorflow/contrib/image/BUILD')
| -rwxr-xr-x[-rw-r--r--] | tensorflow/contrib/image/BUILD | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/tensorflow/contrib/image/BUILD b/tensorflow/contrib/image/BUILD index c31ac7b324..7599406aca 100644..100755 --- a/tensorflow/contrib/image/BUILD +++ b/tensorflow/contrib/image/BUILD @@ -87,6 +87,7 @@ cuda_py_test( srcs = ["python/kernel_tests/image_ops_test.py"], additional_deps = [ ":image_py", + ":single_image_random_dot_stereograms_py", "//third_party/py/numpy", "//tensorflow/python:array_ops", "//tensorflow/python:framework_for_generated_wrappers", @@ -96,6 +97,33 @@ cuda_py_test( ], ) +tf_custom_op_library( + name = "python/ops/_single_image_random_dot_stereograms.so", + srcs = [ + "kernels/single_image_random_dot_stereograms_ops.cc", + "ops/single_image_random_dot_stereograms_ops.cc", + ], +) + +tf_gen_op_libs( + op_lib_names = ["single_image_random_dot_stereograms_ops"], +) + +tf_gen_op_wrapper_py( + name = "single_image_random_dot_stereograms_ops", + deps = [":single_image_random_dot_stereograms_ops_op_lib"], +) + +py_library( + name = "single_image_random_dot_stereograms_py", + srcs = glob(["python/ops/single*.py"]) + ["__init__.py"], + data = [":python/ops/_single_image_random_dot_stereograms.so"], + srcs_version = "PY2AND3", + deps = [ + ":single_image_random_dot_stereograms_ops", + ], +) + filegroup( name = "all_files", srcs = glob( |