code: def read_tf_checkpoint(path):
      
      from tensorflow.python import pywrap_tensorflow
      tensors = {}
      reader = pywrap_tensorflow.NewCheckpointReader(path)
      var_to_shape_map = reader.get_variable_to_shape_map()
      for key in sorted(var_to_shape_map):
          tensor = reader.get_tensor(key)
          tensors[key] = tensor
      return tensors
docs: read tensorflow checkpoint