code: def testPmfCorrectBroadcastDynamicShape(self):
      p = tf1.placeholder_with_default([0.2, 0.3, 0.4], shape=None)
      dist = probit_bernoulli.ProbitBernoulli(probs=p)
      event1 = [1, 0, 1]
      event2 = [[1, 0, 1]]
      self.assertAllClose(
          [0.2, 0.7, 0.4], self.evaluate(dist.prob(event1))
      )
      self.assertAllClose(
          [[0.2, 0.7, 0.4]], self.evaluate(dist.prob(event2))
      )
  
imports: import numpy
  from scipy import special
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions import probit_bernoulli
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 8 for calling the ProbitBernoulliTest obj's underlying member method code with example usage: obj.testPmfCorrectBroadcastDynamicShape() without return types