code: def ops(input, weight):
      return (input.mm(weight)).mm(weight).sum()
  
imports: import unittest
  import functools
  import itertools
  from apex import amp
  from apex.amp import _amp_state
  import torch
  from torch import nn
  import torch.nn.functional
  from utils import common_init, HALF, FLOAT, ALWAYS_HALF, ALWAYS_FLOAT, MATCH_INPUT
label_desc: Use image node_id 2 for calling the WhitelistModule obj's underlying member method code with example usage: obj.ops(input, weight) and returns: input