code: def test_load_with_partitioner_works(self):
      model = self.Model()
      model_dir = self.get_temp_dir()
      tf.saved_model.save(model, model_dir)
  
      strategy = parameter_server_strategy_v2.ParameterServerStrategyV2(
          self.cluster_resolver, tf1.fixed_size_partitioner(2)
      )
      with strategy.scope():
          tf.saved_model.load(model_dir)
  
imports: import os
  from absl.testing import parameterized
  import tensorflow
  import tensorflow.compat.v1
  from tensorflow.python.distribute import combinations
  from tensorflow.python.distribute import multi_worker_test_base
  from tensorflow.python.distribute import parameter_server_strategy_v2
  from tensorflow.python.distribute import sharded_variable
  from tensorflow.python.distribute import strategy_combinations
  from tensorflow.python.distribute import test_util
  from tensorflow.python.distribute import values
  from tensorflow.python.eager import context
  from tensorflow.python.eager import test
  from tensorflow.python.ops import lookup_ops
label_desc: Use image node_id 8 for calling the PSStrategySaveAndLoadTest obj's underlying member method code with example usage: obj.test_load_with_partitioner_works() without return types