code: def _check_initial_dtypes(var_defs, init_vals, backend):
      var_def_dict = dict(var_defs)
      for varname, val in six.iteritems(init_vals):
          for tensor_type, subval in inst.pattern_zip(
              var_def_dict[varname].tensors,
              val,
              leaf_type=inst.TensorType,
          ):
              backend.assert_matching_dtype(
                  tensor_type.dtype,
                  subval,
                  "var name {}".format(varname),
              )
  
file_path: probability/tensorflow_probability/python/experimental/auto_batching/virtual_machine.py
imports: import contextlib
  import functools
  from absl import logging
  import six
  from tensorflow_probability.python.experimental.auto_batching import instructions
label_desc: Use image node_id 8 for calling a global function with example usage: _check_initial_dtypes(var_defs, init_vals, backend) without return types