code: def _dtypes_to_test(use_gpu):
      # Based on issue #347 (https://github.com/tensorflow/addons/issues/347)
      # tf.half is not registered for 'ResourceScatterUpdate' OpKernel for 'GPU'.
      # So we have to remove tf.half when testing with gpu.
      if use_gpu:
          return [tf.float32, tf.float64]
      else:
          return [tf.half, tf.float32, tf.float64]
  
imports: import numpy
  from numpy import linalg
  import pytest
  import tensorflow
  from tensorflow_addons.optimizers import lamb
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 1 for calling a global function with example usage: _dtypes_to_test(use_gpu) and returns: list, list