diff --git a/tensorflow/python/distribute/values.py b/tensorflow/python/distribute/values.py
index 4c64041d2db..ecd29197b7b 100644
--- a/tensorflow/python/distribute/values.py
+++ b/tensorflow/python/distribute/values.py
@@ -282,10 +282,7 @@ class DistributedValues(object):
 
   @property
   def is_tensor_like(self):
-    for v in self._values:
-      if not tensor_util.is_tensor(v):
-        return False
-    return True
+    return all(tensor_util.is_tensor(v) for v in self._values)
 
   def __str__(self):
     devices = self.devices