code: def _log1p_pade_4_4_jvp(x, g):
      return _log1p_pade_4_4_fwd(x[0])[0], _log1p_pade_4_4_bwd(
          x[0], g[0]
      )
  
imports: import contextlib
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import custom_gradient
  from tensorflow_probability.python.internal import dtype_util
label_desc: Use image node_id 21 for calling a global function with example usage: _log1p_pade_4_4_jvp(x, g) and returns: _log1p_pade_4_4_fwd,  _log1p_pade_4_4_bwd