code: def _check_shape_dtype(shape_dtype):
      dt = np.dtype(shape_dtype.dtype)
      if dtypes.canonicalize_dtype(dt) != dt:
          raise ValueError(
              "Cannot return 64-bit values when `jax_enable_x64` is disabled"
          )
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  import functools
  from typing import Any, Callable
  import numpy
  from jax._src import core
  from jax._src import dispatch
  from jax._src import dtypes
  from jax._src import effects
  from jax._src import sharding_impls
  from jax._src import tree_util
  from jax._src import util
  from jax._src.interpreters import ad
  from jax._src.interpreters import batching
  from jax._src.interpreters import mlir
  from jax._src.lib import xla_client
  from jax._src.lax.control_flow.loops import map
  from jax._src.sharding_impls import SingleDeviceSharding
label_desc: Use image node_id 8 for calling a global function with example usage: _check_shape_dtype(shape_dtype) without return types