code: def test_dtype(self, dtype):
      if dtype == tf.float16 and self.primary_device == "TPU":
          self.skipTest("float16 embeddings not supported on TPU.")
      e = embed.Embed(1, dtype=dtype)
      self.assertEqual(e.embeddings.dtype, dtype)
  
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 8 for calling the EmbedTest obj's underlying member method code with example usage: obj.test_dtype(dtype) without return types