From 33512798366dc45068b7cd1bf3fcb9ac1b82c4fb Mon Sep 17 00:00:00 2001 From: Gaurav Jain Date: Wed, 31 Jul 2019 15:54:43 -0700 Subject: [PATCH] Avoid doing an equality check on Tensor dimensions PiperOrigin-RevId: 261015435 --- tensorflow/python/ops/special_math_ops.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/ops/special_math_ops.py b/tensorflow/python/ops/special_math_ops.py index 7ad841c18a4..2f350c114fd 100644 --- a/tensorflow/python/ops/special_math_ops.py +++ b/tensorflow/python/ops/special_math_ops.py @@ -533,7 +533,8 @@ def _reshape_if_necessary(tensor, new_shape): new_shape = tuple(-1 if x is None else x for x in new_shape) cur_shape = tuple(x.value for x in tensor.get_shape().dims) if (len(new_shape) == len(cur_shape) and - all(d0 == d1 or d1 == -1 for d0, d1 in zip(cur_shape, new_shape))): + all(not isinstance(d1, ops.Tensor) and (d0 == d1 or d1 == -1) + for d0, d1 in zip(cur_shape, new_shape))): return tensor else: return array_ops.reshape(tensor, new_shape)