code: def solarize_add(data, shift):
      mask_shifted = data < types.Constant(128, dtype=types.UINT8)
      mask_id = mask_shifted ^ True
      sample_shifted = data + shift
      return mask_shifted * sample_shifted + mask_id * data
  
imports: from nvidia.dali import fn
  from nvidia.dali import types
  from nvidia.dali.auto_aug.core import augmentation
label_desc: Use image node_id 21 for calling a global function with example usage: solarize_add(data, shift) and returns: unknown