code: def forward(ctx, bias, activation, *args):
      output = mlp_cuda.forward(bias, activation, args)
      ctx.save_for_backward(*args)
      ctx.outputs = output
      ctx.bias = bias
      ctx.activation = activation
      return output[0]
  
file_path: apex/apex/mlp/mlp.py
imports: from copy import copy
  import math
  import torch
  from torch import nn
  from apex._autocast_utils import _cast_if_autocast_enabled
  import mlp_cuda
label_desc: Use image node_id 1 for calling the MlpFunction obj's underlying member method code with example usage: obj.forward(ctx, bias, activation) and returns: output