code: def compute_quantities_for_vmap_test(
      op,
      orig_batched_args,
      orig_kwarg_values,
      in_dims,
      out_dim=0,
      batch_size=2,
      compute_loop_out=True,
      clone_inputs=False,
  ):
      for quantities in _compute_quantities_for_vmap_test(
          op,
          orig_batched_args,
          orig_kwarg_values,
          in_dims,
          out_dim,
          batch_size,
          compute_loop_out,
          clone_inputs,
      ):
          yield (quantities[0], quantities[1])
          yield (quantities[2], quantities[3])
  
imports: import itertools
  import torch
  from functorch import vmap
  import torch.utils._pytree
  from functorch_additional_op_db import additional_op_db
  from torch.testing._internal.common_methods_invocations import DecorateInfo
  from torch.testing._internal.common_methods_invocations import op_db
  from torch.testing._internal.common_modules import module_db
  import os
  import unittest
  from torch.testing._internal.common_device_type import toleranceOverride
  from torch.testing._internal.autograd_function_db import autograd_function_db
  from collections import namedtuple
label_desc: Use image node_id 13 for calling a global function with example usage: compute_quantities_for_vmap_test(op, orig_batched_args, orig_kwarg_values, in_dims, out_dim, batch_size, compute_loop_out, clone_inputs) without return types