code: def forward(self, x):
      x = self.first_conv(x)
      for block in self.blocks:
          x = block(x)
      x = self.feature_mix_layer(x)
      x = self.global_avg_pooling(x)
      x = x.view(x.size(0), -1)
      x = self.classifier(x)
      return x
  
file_path: nni/examples/nas/legacy/oneshot/proxylessnas/model.py
imports: import torch
  import nni.retiarii.nn.pytorch
  import math
  import ops
  import putils
  from nni.retiarii.nn.pytorch import LayerChoice
label_desc: Use image node_id 2 for calling the SearchMobileNet obj's underlying member method code with example usage: obj.forward(x) and returns: x