code: def _log_pade_4_4_jvp(x, dx):
      return (
          _log_pade_4_4_fwd(x[0])[0],
          _log_pade_4_4_bwd(x[0], dx[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 11 for calling a global function with example usage: _log_pade_4_4_jvp(x, dx) and returns: _log_pade_4_4_fwd,  _log_pade_4_4_bwd