code: def restore_stacks_and_parts(trackables, merged_prefix):
      (
          tensor_names,
          shapes_and_slices,
          tensors,
          restored_trackables,
      ) = get_tensor_slices(trackables)
      dtypes = [t.dtype for t in tensors]
      restored_tensors = io_ops.restore_v2(
          merged_prefix, tensor_names, shapes_and_slices, dtypes
      )
      for trackable, restored_tensor in zip(
          restored_trackables, restored_tensors
      ):
          expected_shape = trackable.value().get_shape()
          restored_tensor = array_ops.reshape(
              restored_tensor, expected_shape
          )
          parts = array_ops_stack.unstack(restored_tensor)
          for part, restored_part in zip(trackable.parts, parts):
              part.assign(restored_part)
  
imports: import os
  import tempfile
  from absl.testing import parameterized
  from google.protobuf import wrappers_pb2
  from tensorflow.python.checkpoint import checkpoint
  from tensorflow.python.client import session
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.eager import test
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import io_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import gfile
  from tensorflow.python.saved_model import load
  from tensorflow.python.saved_model import loader
  from tensorflow.python.saved_model import registration
  from tensorflow.python.saved_model import save
  from tensorflow.python.trackable import autotrackable
label_desc: Use image node_id 3 for calling a global function with example usage: restore_stacks_and_parts(trackables, merged_prefix) without return types