code: def testDtype(self):
      temperature = tf.constant(1.0, dtype=tf.float32)
      p = tf.constant([0.1, 0.4], dtype=tf.float32)
      dist = relaxed_bernoulli.RelaxedBernoulli(
          temperature, probs=p, validate_args=True
      )
      self.assertEqual(dist.dtype, tf.float32)
      self.assertEqual(
          dist.dtype, dist.sample(5, seed=test_util.test_seed()).dtype
      )
      self.assertEqual(dist.probs.dtype, dist.prob([0.0]).dtype)
      self.assertEqual(dist.probs.dtype, dist.log_prob([0.0]).dtype)
  
      temperature = tf.constant(1.0, dtype=tf.float64)
      p = tf.constant([0.1, 0.4], dtype=tf.float64)
      dist64 = relaxed_bernoulli.RelaxedBernoulli(
          temperature, probs=p, validate_args=True
      )
      self.assertEqual(dist64.dtype, tf.float64)
      self.assertEqual(
          dist64.dtype,
          dist64.sample(5, seed=test_util.test_seed()).dtype,
      )
  
imports: import numpy
  import scipy.special
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import relaxed_bernoulli
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 6 for calling the RelaxedBernoulliTest obj's underlying member method code with example usage: obj.testDtype() without return types