code: def _igamma_grad_a(
      a, x, name=None
  ):  # pylint: disable=unused-argument
      if JAX_MODE:
          import jax.lax  # pylint: disable=g-import-not-at-top
  
          # Broadcast a and x.
  
          a, x = np.broadcast_arrays(a, x)
          return jax.lax.igamma_grad_a(a, x)
      raise NotImplementedError
  
imports: import collections
  import numpy
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
label_desc: Use image node_id 4 for calling a global function with example usage: _igamma_grad_a(a, x, name) and returns: jax