diff options
| author |  Derek Murray <mrry@google.com> | 2016-04-08 06:04:37 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-04-08 07:11:59 -0700 |
| commit | 91969ae68e30eef368698094a7f574a3bcfdf315 (patch) | |
| tree | a457140e07393cbc858944b384a02ac2a2c94879 /tensorflow/python/kernel_tests/reverse_sequence_op_test.py | |
| parent | 6cdcc5945183070d51b85cba79c089416e55a736 (diff) | |
Fixes `tf.reverse_sequence()` to work when input shape is unknown.
Fixes #1816.
Change: 119369338
Diffstat (limited to 'tensorflow/python/kernel_tests/reverse_sequence_op_test.py')
| -rw-r--r-- | tensorflow/python/kernel_tests/reverse_sequence_op_test.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/reverse_sequence_op_test.py b/tensorflow/python/kernel_tests/reverse_sequence_op_test.py index d5b087298e..ccb19009fb 100644 --- a/tensorflow/python/kernel_tests/reverse_sequence_op_test.py +++ b/tensorflow/python/kernel_tests/reverse_sequence_op_test.py @@ -114,6 +114,12 @@ class ReverseSequenceTest(tf.test.TestCase): self.assertLess(err, 1e-8) def testShapeFunctionEdgeCases(self): + t = tf.reverse_sequence( + tf.placeholder(tf.float32, shape=None), + seq_lengths=tf.placeholder(tf.int64, shape=(32,)), + batch_dim=0, seq_dim=1) + self.assertIs(t.get_shape().ndims, None) + # Batch size mismatched between input and seq_lengths. with self.assertRaises(ValueError): tf.reverse_sequence( |