diff options
Diffstat (limited to 'tensorflow/contrib/sparsemax/BUILD')
-rw-r--r-- | tensorflow/contrib/sparsemax/BUILD | 76 |
1 files changed, 76 insertions, 0 deletions
diff --git a/tensorflow/contrib/sparsemax/BUILD b/tensorflow/contrib/sparsemax/BUILD new file mode 100644 index 0000000000..bd59c626f2 --- /dev/null +++ b/tensorflow/contrib/sparsemax/BUILD @@ -0,0 +1,76 @@ +# Description: +# Contains ops to train linear models on top of TensorFlow. +# APIs here are meant to evolve over time. + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +package(default_visibility = ["//visibility:public"]) + +load("//tensorflow:tensorflow.bzl", "cuda_py_tests") +load( + "//tensorflow:tensorflow.bzl", + "tf_custom_op_library", + "tf_py_test", +) +load( + "//tensorflow/core:platform/default/build_config.bzl", + "tf_kernel_tests_linkstatic", +) + +py_library( + name = "sparsemax_py", + srcs = ["__init__.py"] + glob(["python/ops/*.py"]), + srcs_version = "PY2AND3", + visibility = ["//visibility:public"], + deps = [ + "//tensorflow/contrib/util:util_py", + "//tensorflow/python:array_ops", + "//tensorflow/python:framework_for_generated_wrappers", + "//tensorflow/python:math_ops", + "//tensorflow/python:nn", + ], +) + +cuda_py_tests( + name = "sparsemax_test", + size = "small", + srcs = ["python/kernel_tests/sparsemax_test.py"], + additional_deps = [ + ":sparsemax_py", + "//tensorflow:tensorflow_py", + "//tensorflow/python:array_ops", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:gradients", + "//tensorflow/python:math_ops", + "//tensorflow/python:platform_test", + ], +) + +cuda_py_tests( + name = "sparsemax_loss_test", + size = "medium", + srcs = ["python/kernel_tests/sparsemax_loss_test.py"], + additional_deps = [ + ":sparsemax_py", + "//tensorflow:tensorflow_py", + "//tensorflow/python:array_ops", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:gradients", + "//tensorflow/python:math_ops", + "//tensorflow/python:platform_test", + ], +) + +filegroup( + name = "all_files", + srcs = glob( + ["**/*"], + exclude = [ + "**/METADATA", + "**/OWNERS", + ], + ), + visibility = ["//tensorflow:__subpackages__"], +) |