diff options
author | 2018-01-26 16:36:55 -0800 | |
---|---|---|
committer | 2018-01-26 16:40:31 -0800 | |
commit | a977a77299f292e556ace48c75251a5a11d118ff (patch) | |
tree | 0cbcaaa6999f5d0833374de979b983615fc083d4 /tensorflow/contrib/lite/kernels/BUILD | |
parent | e9dc418b4bafa359654fd66c72b5f4ba371b68db (diff) |
Add bidirectional sequence RNN to TFLite Ops.
PiperOrigin-RevId: 183465032
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/kernels/BUILD | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD index b5428d3246..d9051f3516 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -104,6 +104,7 @@ cc_library( "add.cc", "basic_rnn.cc", "batch_to_space_nd.cc", + "bidirectional_sequence_rnn.cc", "concatenation.cc", "conv.cc", "depthwise_conv.cc", @@ -289,6 +290,18 @@ tf_cc_test( ) tf_cc_test( + name = "bidirectional_sequence_rnn_test", + size = "small", + srcs = ["bidirectional_sequence_rnn_test.cc"], + deps = [ + ":builtin_ops", + "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite/kernels:test_util", + "@com_google_googletest//:gtest", + ], +) + +tf_cc_test( name = "unidirectional_sequence_rnn_test", size = "small", srcs = ["unidirectional_sequence_rnn_test.cc"], |