code: def _log_cosh_jvp(primals, tangents):
      (x,) = primals
      (dx,) = tangents
      return _log_cosh_impl(x), tf.math.tanh(x) * dx
  
imports: import collections
  import functools
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import custom_gradient
  from tensorflow_probability.python.internal import distribute_lib
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import variadic_reduce
  from tensorflow_probability.python.math.scan_associative import scan_associative
  from tensorflow.python.util import deprecation
label_desc: Use image node_id 18 for calling a global function with example usage: _log_cosh_jvp(primals, tangents) and returns: _log_cosh_impl,  unknown