code: def _split_fn_calls(instructions):
      """Yields lists of straight-line ops and individual `FunctionCallOp`s."""
      pending = []
      for op in instructions:
          if isinstance(op, inst.FunctionCallOp):
              if pending:
                  yield pending
                  pending = []
              yield op
          else:
              pending.append(op)
      if pending:
          yield pending
  
file_path: probability/tensorflow_probability/python/experimental/auto_batching/stackless.py
imports: import contextlib
  import functools
  import heapq
  from absl import logging
  import six
  from tensorflow_probability.python.experimental.auto_batching import instructions
label_desc: Use image node_id 2 for calling a global function with example usage: _split_fn_calls(instructions) without return types