code: def _distribution_strategies():
      return [
          tf.distribute.experimental.MultiWorkerMirroredStrategy(),
          tf.distribute.MirroredStrategy(),
          # TODO(pulkitb): Add parameter_server
          # tf.distribute.experimental.ParameterServerStrategy once ready in TF2.
          tf.distribute.OneDeviceStrategy("/cpu:0"),
      ]
  
file_path: model-optimization/tensorflow_model_optimization/python/core/sparsity/keras/prune_distributed_test.py
imports: import tempfile
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.keras import test_utils
  from tensorflow_model_optimization.python.core.sparsity.keras import prune
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_callbacks
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_schedule
  from tensorflow_model_optimization.python.core.sparsity.keras import test_utils
label_desc: Use image node_id 1 for calling a global function with example usage: _distribution_strategies() and returns: list