code: def rwb_pbroadcast(x, axis_name):
      """Applies a pbroadcast with reduce-without-broadcast (RWB) semantics."""
      return _rwb_pbroadcast_fwd(x, axis_name)[0]
  
imports: import collections
  import functools
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import custom_gradient
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import samplers
  from tensorflow.python.util import nest
label_desc: Use image node_id 17 for calling a global function with example usage: rwb_pbroadcast(x, axis_name) and returns: _rwb_pbroadcast_fwd