code: def _dtype(x):
      if hasattr(x, "dtype"):
          return x.dtype
      elif type(x) in _dtypes.python_scalar_dtypes:
          return np.dtype(_dtypes.python_scalar_dtypes[type(x)])
      else:
          return np.asarray(x).dtype
  
imports: from functools import partial
  import operator
  from jax._src import api
  from jax._src import config
  from jax._src import dtypes
  from jax._src.tree_util import tree_map, tree_reduce
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: _dtype(x) and returns: x, np, np