code: def reduce_input(input_, parallel_mode):
      return _ReduceInput.apply(input_, parallel_mode)
  
imports: import torch
  import torch.distributed
  from colossalai.legacy.core import global_context
  from colossalai.legacy.global_variables import tensor_parallel_env
  from ..utils import divide
label_desc: Use image node_id 9 for calling a global function with example usage: reduce_input(input_, parallel_mode) and returns: _ReduceInput