code: def test_sparse(dtype):
      # TODO: remove the with tf.device when the execution on cpu is enforced
      # See #1682 to track it.
      with tf.device("CPU:0"):
          _test_sparse(dtype)
  
file_path: addons/tensorflow_addons/optimizers/tests/lazy_adam_test.py
imports: import numpy
  import tensorflow
  from tensorflow_addons.optimizers import lazy_adam
  from tensorflow_addons.utils import test_utils
  import pytest
label_desc: Use image node_id 3 for calling a global function with example usage: test_sparse(dtype) without return types