code: def _dtypes_to_test(use_gpu):
      if use_gpu:
          return [tf.dtypes.float32, tf.dtypes.float64]
      else:
          return [tf.dtypes.half, tf.dtypes.float32, tf.dtypes.float64]
  
imports: import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.optimizers import yogi
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 3 for calling a global function with example usage: _dtypes_to_test(use_gpu) and returns: list, list