From fa5ad62d329088a3ec397c8bd8fc78001960034a Mon Sep 17 00:00:00 2001 From: Alexandre Passos Date: Thu, 4 Apr 2019 09:35:14 -0700 Subject: [PATCH] Proper check for eagerness in concat_grad PiperOrigin-RevId: 241945534 --- tensorflow/python/ops/array_grad.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/ops/array_grad.py b/tensorflow/python/ops/array_grad.py index 867a4b7b3c8..cbb811054af 100644 --- a/tensorflow/python/ops/array_grad.py +++ b/tensorflow/python/ops/array_grad.py @@ -107,7 +107,7 @@ def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index): out_grads = [] if isinstance(grad, ops.Tensor): - if context.executing_eagerly(): + if context.executing_eagerly() or isinstance(concat_dim, ops.EagerTensor): # Using mod here for convenience since concat_dim is already verified # in concat implementation to be within the allowed [-rank, rank) range. non_neg_concat_dim = (