code: def setUp(self):
      super(PackedDistributedVariableTest, self).setUp()
      cpus = config.list_physical_devices("CPU")
      # Set 2 virtual CPUs
      config.set_logical_device_configuration(
          cpus[0],
          [
              context.LogicalDeviceConfiguration(),
              context.LogicalDeviceConfiguration(),
          ],
      )
  
file_path: tensorflow-upstream/tensorflow/python/distribute/packed_distributed_variable_test.py
imports: from tensorflow.python.distribute import device_util
  from tensorflow.python.distribute import packed_distributed_variable
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import config
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the PackedDistributedVariableTest obj's underlying member method code with example usage: obj.setUp() without return types