code: def _variable_crosses_block_boundary(varname, alloc):
      if (
          alloc[varname] is inst.VariableAllocation.NULL
          or alloc[varname] is inst.VariableAllocation.TEMPORARY
      ):
          alloc[varname] = inst.VariableAllocation.REGISTER
  
file_path: probability/tensorflow_probability/python/experimental/auto_batching/allocation_strategy.py
imports: from absl import logging
  import six
  from tensorflow_probability.python.experimental.auto_batching import instructions
  from tensorflow_probability.python.experimental.auto_batching import liveness
label_desc: Use image node_id 4 for calling a global function with example usage: _variable_crosses_block_boundary(varname, alloc) without return types