code: def testDTypes(self, dtype):
      # check that sampling and log_prob work for a range of dtypes
      logits = tf.random.uniform(
          shape=[3, 3], dtype=dtype, seed=test_util.test_seed()
      )
      dist = roc.RelaxedOneHotCategorical(
          temperature=0.5, logits=logits, validate_args=True
      )
      dist.log_prob(dist.sample(seed=test_util.test_seed()))
  
imports: from absl.testing import parameterized
  import numpy
  from scipy.special import gamma
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import identity
  from tensorflow_probability.python.distributions import relaxed_onehot_categorical
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.util import deferred_tensor
label_desc: Use image node_id 12 for calling the RelaxedOneHotCategoricalTest obj's underlying member method code with example usage: obj.testDTypes(dtype) without return types