code: def add_inshape(module_masks, mask):
      """
      Inference the output mask of the add operation from the
      input mask.
      """
      assert isinstance(mask, CoarseMask)
      if module_masks.input_mask is None:
          module_masks.set_input_mask(mask)
          module_masks.set_output_mask(mask)
          # module_masks.input_mask = mask
          return mask
      # If alreay visited, validate if have the conflict
      # if the mask is different with previous input_mask
      # then there is a mask confilct.
      if mask != module_masks.input_mask:
          raise Exception("Mask conflict happenes!")
      return None
  
imports: import logging
  import torch
label_desc: Use image node_id 5 for calling a global function with example usage: add_inshape(module_masks, mask) and returns: None, mask