code: def test_custom_device_placement():
      with tf.device(test_utils.gpus_for_testing()[0]):
          train_small_model()
  
      strategy = tf.distribute.MirroredStrategy(
          test_utils.gpus_for_testing()
      )
      with strategy.scope():
          train_small_model()
  
file_path: addons/tensorflow_addons/utils/tests/test_utils_test.py
imports: import random
  from packaging.version import Version
  import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 5 for calling a global function with example usage: test_custom_device_placement() without return types