code: def conv_dw(inp, oup, stride):
      return nn.Sequential(
          nn.Conv2d(inp, inp, 3, stride, 1, groups=inp, bias=False),
          nn.BatchNorm2d(inp),
          nn.ReLU(inplace=True),
          nn.Conv2d(inp, oup, 1, 1, 0, bias=False),
          nn.BatchNorm2d(oup),
          nn.ReLU(inplace=True),
      )
  
file_path: nni/test/ut/sdk/models/pytorch_models/mobilenet.py
imports: import torch.nn
  import math
label_desc: Use image node_id 2 for calling a global function with example usage: conv_dw(inp, oup, stride) and returns: nn