Make output of fn in @recompute_grad a list to avoid trying to concat…

… tuple and list PiperOrigin-RevId: 169632380
tensorflow · Sep 22, 2017 · 76706ef · 76706ef
1 parent 6237729
commit 76706ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensor2tensor/layers/rev_block.py b/tensor2tensor/layers/rev_block.py
@@ -348,7 +348,7 @@ def _recompute_grad(fn, args):
   def grad_fn(inputs, variables, outputs, output_grads):
     del outputs
     # recompute outputs
-    outputs = fn(*inputs)
+    outputs = list(fn(*inputs))
     grads = tf.gradients(outputs, inputs + variables, output_grads)
     grad_inputs = grads[:len(inputs)]
     grad_vars = grads[len(inputs):]