code: def update_num_microbatches(consumed_samples, consistency_check=True):
      _GLOBAL_NUM_MICROBATCHES_CALCULATOR.update(
          consumed_samples, consistency_check
      )
  
imports: from typing import Optional, List, Union, Tuple
  import torch
  from torch.nn.parallel import DistributedDataParallel
  from apex.multi_tensor_apply import multi_tensor_applier
  from apex.transformer import parallel_state
  from apex.transformer.enums import ModelType
  from apex.transformer.microbatches import build_num_microbatches_calculator
  from apex.transformer.pipeline_parallel._timers import _Timers
label_desc: Use image node_id 9 for calling a global function with example usage: update_num_microbatches(consumed_samples, consistency_check) without return types