code: def _input_dtype(x, *_, **__):
      return dtypes.canonicalize_dtype(
          x.dtype, allow_extended_dtype=True
      )
  
file_path: jax/jax/_src/lax/utils.py
imports: from functools import partial
  import operator
  from jax._src import core
  from jax._src import dispatch
  from jax._src import dtypes
  from jax._src.util import safe_zip
  from jax._src.lib import xla_client
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: _input_dtype(x) and returns: dtypes