code: def test_pinned_to_cpu(self):
      with tf.device("CPU"):
          e = embed.Embed(1)
      spec = tf.DeviceSpec.from_string(e.embeddings.device)
      self.assertEqual(spec.device_type, "CPU")
  
file_path: sonnet/sonnet/src/embed_test.py
imports: from absl.testing import parameterized
  from sonnet.src import embed
  from sonnet.src import initializers
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 6 for calling the EmbedTest obj's underlying member method code with example usage: obj.test_pinned_to_cpu() without return types