aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/BUILD')
-rw-r--r--tensorflow/python/BUILD17
1 files changed, 17 insertions, 0 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD
index 9c0db82cbf..524f128154 100644
--- a/tensorflow/python/BUILD
+++ b/tensorflow/python/BUILD
@@ -3785,6 +3785,7 @@ py_library(
"layers/convolutional.py",
"layers/core.py",
"layers/layers.py",
+ "layers/maxout.py",
"layers/normalization.py",
"layers/pooling.py",
],
@@ -3866,6 +3867,22 @@ py_test(
)
py_test(
+ name = "layers_maxout_test",
+ size = "small",
+ srcs = ["layers/maxout_test.py"],
+ main = "layers/maxout_test.py",
+ srcs_version = "PY2AND3",
+ deps = [
+ ":client_testlib",
+ ":framework_for_generated_wrappers",
+ ":layers",
+ ":math_ops",
+ ":nn_ops",
+ ":random_ops",
+ ],
+)
+
+py_test(
name = "layers_utils_test",
size = "small",
srcs = ["layers/utils_test.py"],