diff options
author | 2016-10-06 13:34:23 -0800 | |
---|---|---|
committer | 2016-10-06 14:47:08 -0700 | |
commit | 4fe49431eb892184c03e4af57d9a0b36b1af6989 (patch) | |
tree | f415db8947aa8658a9d70e39d3ecfa40d8727719 /tensorflow/contrib/specs/BUILD | |
parent | d680b19889fa7cb6c25a3a32006199e542b3d411 (diff) |
Adding specs to the open source version of TensorFlow.
Change: 135406004
Diffstat (limited to 'tensorflow/contrib/specs/BUILD')
-rw-r--r-- | tensorflow/contrib/specs/BUILD | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/tensorflow/contrib/specs/BUILD b/tensorflow/contrib/specs/BUILD new file mode 100644 index 0000000000..517fe4784c --- /dev/null +++ b/tensorflow/contrib/specs/BUILD @@ -0,0 +1,60 @@ +# Description: +# A small domain-specific language (DSL) for defining deep learning networks. + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +package(default_visibility = ["//tensorflow:__subpackages__"]) + +load("//tensorflow:tensorflow.bzl", "tf_py_test") + +py_library( + name = "specs", + srcs = [ + "python/__init__.py", + "python/params_ops.py", + "python/specs.py", + "python/specs_lib.py", + "python/specs_ops.py", + "python/summaries.py", + ], + srcs_version = "PY2AND3", + deps = [ + "//tensorflow/contrib/ndlstm", + "//tensorflow/python:framework", + "//tensorflow/python:ops", + "//tensorflow/python:platform", + "//tensorflow/python:training", + ], +) + +tf_py_test( + name = "specs_test", + srcs = ["python/specs_test.py"], + additional_deps = [ + ":specs", + "//tensorflow:tensorflow_py", + ], +) + +tf_py_test( + name = "summaries_test", + srcs = ["python/summaries_test.py"], + additional_deps = [ + ":specs", + "//tensorflow:tensorflow_py", + ], +) + +filegroup( + name = "all_files", + srcs = glob( + ["**/*"], + exclude = [ + "**/METADATA", + "**/OWNERS", + ], + ), + visibility = ["//tensorflow:__subpackages__"], +) |