code: def test_static(self):
      self.assertAllEqual(
          (3, 4, 2, 5), ps.broadcast_shape((4, 1, 1), (3, 1, 2, 5))
      )
  
      self.assertAllEqual(
          (3, 4, 2, 5),
          ps.broadcast_shape(
              tf.convert_to_tensor((4, 1, 1)),
              tf.convert_to_tensor((3, 1, 2, 5)),
          ),
      )
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 1 for calling the BroadcastShapeTest obj's underlying member method code with example usage: obj.test_static() without return types