code: def batch_shape_parts(
      batch_object, bijector_x_event_ndims=None, **parameter_kwargs
  ):
      """Returns a dict mapping parameter names to their inferred batch shapes."""
      return map_fn_over_parameters_with_event_ndims(
          batch_object,
          get_batch_shape_tensor_part,
          bijector_x_event_ndims=bijector_x_event_ndims,
          **parameter_kwargs,
      )
  
file_path: probability/tensorflow_probability/python/internal/batch_shape_lib.py
imports: import functools
  from absl import logging
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import tensor_util
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow.python.util import nest
label_desc: Use image node_id 7 for calling a global function with example usage: batch_shape_parts(batch_object, bijector_x_event_ndims) and returns: map_fn_over_parameters_with_event_ndims