diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2016-09-30 01:55:48 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-30 03:04:17 -0700 |
commit | e355ce3289c5b12e226967af95bbebf4a1057e2c (patch) | |
tree | c8f0675dd816ebd9268a86b30c82420f2689a541 /tensorflow/contrib/rnn/BUILD | |
parent | e1074f596dd42481c055b87e78c8f64d02d4c7b7 (diff) |
Helper functions to build multilayer bidirectional rnn.
Using bidirectional_rnn with MultiRNN cells creates two independent forward and backward networks whose output is only combined in the last layer. With the stack_bidirectional_rnn the outputs of both nets are combined and fed as input of the next layer.
Change: 134764000
Diffstat (limited to 'tensorflow/contrib/rnn/BUILD')
-rw-r--r-- | tensorflow/contrib/rnn/BUILD | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/contrib/rnn/BUILD b/tensorflow/contrib/rnn/BUILD index e266055883..e541d9ec38 100644 --- a/tensorflow/contrib/rnn/BUILD +++ b/tensorflow/contrib/rnn/BUILD @@ -67,6 +67,18 @@ cuda_py_tests( ], ) +cuda_py_tests( + name = "rnn_test", + size = "medium", + srcs = ["python/kernel_tests/rnn_test.py"], + additional_deps = [ + ":rnn_py", + "//tensorflow:tensorflow_py", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:platform_test", + ], +) + tf_custom_op_library( name = "python/ops/_lstm_ops.so", srcs = [ |