diff options
author | A. Unique TensorFlower <nobody@tensorflow.org> | 2016-03-24 08:19:12 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-03-24 09:33:05 -0700 |
commit | 5b5b8412f0684a548e1e9001421e5d095cda0142 (patch) | |
tree | 88d3c7419498f7a6991113bac146320b833b601b /tensorflow/contrib/losses/BUILD | |
parent | c404448d3b1e44fddc2d6e1c6da9862443112721 (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/BUILD | 42 |
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__"], +) |