code: def depthwise_conv(
      i: int,
      o: int,
      kernel_size: int,
      stride: int = 1,
      padding: int = 0,
      bias: bool = False,
  ) -> nn.Conv2d:
      return nn.Conv2d(
          i, o, kernel_size, stride, padding, bias=bias, groups=i
      )
  
imports: from functools import partial
  from typing import Any, Callable, List, Optional
  import torch
  import torch.nn
  from torch import Tensor
  from ..transforms._presets import ImageClassification
  from ..utils import _log_api_usage_once
  from ._api import register_model, Weights, WeightsEnum
  from ._meta import _IMAGENET_CATEGORIES
  from ._utils import _ovewrite_named_param, handle_legacy_interface
label_desc: Use image node_id 2 for calling the InvertedResidual obj's underlying member method code with example usage: obj.depthwise_conv(i, o, kernel_size, stride, padding, bias) and returns: nn