code: def jit_dropout_add(x, residual, prob, is_training):
      # type: (Tensor, Tensor, float, bool) -> Tensor
      out = F.dropout(x, p=prob, training=True)
      out = residual + out
      return out
  
file_path: apex/apex/contrib/multihead_attn/self_multihead_attn.py
imports: import math
  import torch
  from torch import nn
  from torch.nn import Parameter
  import torch.nn.functional
  from .self_multihead_attn_func import self_attn_func
  from .fast_self_multihead_attn_func import fast_self_attn_func
  from .fast_self_multihead_attn_norm_add_func import fast_self_attn_norm_add_func
  from apex.normalization.fused_layer_norm import FusedLayerNorm
label_desc: Use image node_id 1 for calling a global function with example usage: jit_dropout_add(x, residual, prob, is_training) and returns: out