code: def _collect_partitioned_variable(name, all_vars):
      """Returns list of `tf.Variable` that comprise the partitioned variable."""
      if name + "/part_0" in all_vars:
          var = []
          i = 0
          while name + "/part_%d" % i in all_vars:
              var.append(all_vars[name + "/part_%d" % i])
              i += 1
          return var
      return None
  
imports: from collections import abc
  import os
  import time
  from tensorflow.python.checkpoint import checkpoint_management
  from tensorflow.python.distribute import distribute_lib
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import io_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.ops import variable_scope
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import gfile
  from tensorflow.python.platform import tf_logging
  from tensorflow.python.training import py_checkpoint_reader
  from tensorflow.python.training.saving import saveable_object_util
  from tensorflow.python.util.tf_export import tf_export
label_desc: Use image node_id 12 for calling a global function with example usage: _collect_partitioned_variable(name, all_vars) and returns: None, var