Merge pull request #38299 from marload:patch-1

PiperOrigin-RevId: 309033760
Change-Id: Ie592456212645442a0f5ba2b6891131503a2b835

tensorflow/python/training/warm_starting_util.py

@@ -363,10 +363,8 @@ def _get_grouped_variables(vars_to_warm_start):
   # out the list.
   grouped_variables = {}
   for v in list_of_vars:
-    if not isinstance(v, list):
-      var_name = _infer_var_name([v])
-    else:
-      var_name = _infer_var_name(v)
+    t = [v] if not isinstance(v, list) else v
+    var_name = _infer_var_name(t)
     grouped_variables.setdefault(var_name, []).append(v)
 
   return grouped_variables