diff options
Diffstat (limited to 'tensorflow/python/framework/subscribe.py')
-rw-r--r-- | tensorflow/python/framework/subscribe.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/framework/subscribe.py b/tensorflow/python/framework/subscribe.py index 7797d991da..cee7398974 100644 --- a/tensorflow/python/framework/subscribe.py +++ b/tensorflow/python/framework/subscribe.py @@ -47,7 +47,7 @@ def _recursive_apply(tensors, apply_fn): tensors_type = type(tensors) if tensors_type is ops.Tensor: return apply_fn(tensors) - elif tensors_type is variables.Variable: + elif isinstance(tensors, variables.Variable): return apply_fn(tensors.value()) elif isinstance(tensors, (list, tuple)): tensors = [_recursive_apply(t, apply_fn) for t in tensors] |