import tensorflow as tf

      # The qdist will be defined with no cutoffs,
      qdist = distributions.QuantizedDistribution(
          base_dist_cls=distributions.Uniform,
          lower_cutoff=None,
          upper_cutoff=None,
          a=tf.zeros(
              batch_shape, dtype=tf.float32),
          b=10 * tf.ones(
              batch_shape, dtype=tf.float32))

      # x is random integers in {-3,...,12}.
      x = self._rng.randint(-3, 13, size=batch_shape).astype(np.float32)

      # pmf
