code: def _distributed_dataset():
      cpus = tf.config.experimental.list_physical_devices("CPU")
      tf.config.experimental.set_virtual_device_configuration(
          cpus[0],
          [tf.config.experimental.VirtualDeviceConfiguration()] * 2,
      )
  
      strat = tf.distribute.MirroredStrategy()
      ds = tf.data.Dataset.from_tensor_slices(
          tf.reshape(tf.range(12), (3, 4))
      )
  
      return strat, strat.experimental_distribute_dataset(ds)
  
imports: import itertools
  from absl.testing import parameterized
  import tensorflow
  from tensorflow.python.autograph.tests import reference_test_base
label_desc: Use image node_id 12 for calling a global function with example usage: _distributed_dataset() and returns: strat,  strat