Add gradient for tf.reverse(), fixes #58.

Change: 110608526
tensorflow · Dec 21, 2015 · 1e416cd · 1e416cd
1 parent ef50775
commit 1e416cd
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tensorflow/python/ops/array_grad.py b/tensorflow/python/ops/array_grad.py
@@ -265,3 +265,9 @@ def _ReverseSequenceGrad(op, grad):
                                      seq_dim=op.get_attr("seq_dim"),
                                      seq_lengths=seq_lengths),
           None]
+
+
+@ops.RegisterGradient("Reverse")
+def _ReverseGrad(op, grad):
+  reverse_dims = op.inputs[1]
+  return array_ops.reverse(grad, reverse_dims), None