code: def get_accelerators():
      gpus = tf.config.experimental.list_logical_devices("GPU")
      tpus = tf.config.experimental.list_logical_devices("TPU")
      return [
          tf.DeviceSpec.from_string(d.name).to_string()
          for d in gpus + tpus
      ]
  
file_path: sonnet/sonnet/src/functional/jax_test.py
imports: from absl.testing import parameterized
  from sonnet.src import test_utils
  from sonnet.src.functional import jax
  import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: get_accelerators() and returns: unknown