aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/losses/BUILD
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <nobody@tensorflow.org>2016-03-24 08:19:12 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-03-24 09:33:05 -0700
commit5b5b8412f0684a548e1e9001421e5d095cda0142 (patch)
tree88d3c7419498f7a6991113bac146320b833b601b /tensorflow/contrib/losses/BUILD
parentc404448d3b1e44fddc2d6e1c6da9862443112721 (diff)
In tf.contrib, move framework and loss packages up a level.
Simplify loss ops, only return scalar in public API (see discussion at go/tf_contrib_ops). Add softmax loss. Change: 118034691
Diffstat (limited to 'tensorflow/contrib/losses/BUILD')
-rw-r--r--tensorflow/contrib/losses/BUILD42
1 files changed, 42 insertions, 0 deletions
diff --git a/tensorflow/contrib/losses/BUILD b/tensorflow/contrib/losses/BUILD
new file mode 100644
index 0000000000..8452132c45
--- /dev/null
+++ b/tensorflow/contrib/losses/BUILD
@@ -0,0 +1,42 @@
+# Description:
+# contains parts of TensorFlow that are experimental or unstable and which are not supported.
+
+licenses(["notice"]) # Apache 2.0
+
+exports_files(["LICENSE"])
+
+package(default_visibility = ["//tensorflow:__subpackages__"])
+
+py_library(
+ name = "losses_py",
+ srcs = [
+ "__init__.py",
+ "python/losses/__init__.py",
+ "python/losses/loss_ops.py",
+ ],
+ srcs_version = "PY2AND3",
+)
+
+py_test(
+ name = "loss_ops_test",
+ srcs = glob(["python/losses/loss_ops_test.py"]),
+ srcs_version = "PY2AND3",
+ deps = [
+ ":losses_py",
+ "//tensorflow:tensorflow_py",
+ "//tensorflow/python:framework_test_lib",
+ "//tensorflow/python:platform_test",
+ ],
+)
+
+filegroup(
+ name = "all_files",
+ srcs = glob(
+ ["**/*"],
+ exclude = [
+ "**/METADATA",
+ "**/OWNERS",
+ ],
+ ),
+ visibility = ["//tensorflow:__subpackages__"],
+)