code: def _float_dtype_like(dtype):
      if dtype_util.as_numpy_dtype(dtype) == np.int64:
          return tf.float64
      if dtype_util.is_integer(dtype):
          return tf.float32
      return dtype
  
imports: import functools
  import math
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import nest_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.mcmc import diagnostic
  from tensorflow.python.util import nest
label_desc: Use image node_id 3 for calling a global function with example usage: _float_dtype_like(dtype) and returns: dtype, tf, tf