code: def conv_nxn_bn(inp, oup, kernel_size=3, stride=1):
      return nn.Sequential(
          nn.Conv2d(inp, oup, kernel_size, stride, 1, bias=False),
          nn.BatchNorm2d(oup),
          nn.SiLU(),
      )
  
imports: import torch
  import torch.nn
  from einops import rearrange
  from einops.layers.torch import Reduce
label_desc: Use image node_id 2 for calling a global function with example usage: conv_nxn_bn(inp, oup, kernel_size, stride) and returns: nn