code: def _assert_consumed_value_key_type_signature(eqn, args_consumed):
      actual = args_consumed[0]
      expected = eqn.params["value"]
      if not np.all(actual == expected):
          if np.all(expected):
              raise AssertionError(
                  f"Expected key to be consumed in {eqn}"
              )
          elif not np.any(expected):
              raise AssertionError(
                  f"Expected key to not be consumed in {eqn}"
              )
          else:
              raise AssertionError(
                  f"Expected {expected}, got {actual} in {eqn}"
              )
      return KeyReuseSignature([], [])
  
imports: from __future__ import annotations
  from collections import defaultdict
  from functools import reduce
  from typing import Any, Callable
  import jax
  from jax import core
  from jax import lax
  from jax import tree_util
  from jax._src import api_util
  from jax._src import linear_util
  from jax._src import pjit
  from jax._src import prng
  from jax._src import random
  from jax._src import util
  from jax._src.debugging import debug_callback_p
  from jax._src.interpreters import partial_eval
  from jax.experimental.key_reuse._common import consume_p, unconsumed_copy_p, assert_consumed_value_p, KeyReuseError, Sink, Source, KeyReuseSignature
  import numpy
label_desc: Use image node_id 6 for calling a global function with example usage: _assert_consumed_value_key_type_signature(eqn, args_consumed) and returns: KeyReuseSignature