code: def mlp_function(bias, activation, *args):
      autocast_args = _cast_if_autocast_enabled(bias, activation, *args)
      return MlpFunction.apply(*autocast_args)
  
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 a global function with example usage: mlp_function(bias, activation) and returns: MlpFunction