code: def _check_dtypes_match(xs, ys):
      def _assert_dtypes_match(x, y):
          if config.enable_x64.value:
              assert _dtype(x) == _dtype(y)
          else:
              assert _dtypes.canonicalize_dtype(
                  _dtype(x)
              ) == _dtypes.canonicalize_dtype(_dtype(y))
  
      tree_map(_assert_dtypes_match, xs, ys)
  
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 8 for calling a global function with example usage: _check_dtypes_match(xs, ys) without return types