code: def create_time_mix_layer(
      in_feat,
      out_feat,
      kernel_size=3,
      stride=1,
      conv_type="depth-wise",
      dilation=1,
  ):
      padding = get_same_padding(
          kernel_size, stride=stride, dilation=dilation
      )
  
      if conv_type == "original":
          groups = 1
      elif conv_type == "depth-wise":
          groups = in_feat
      else:
          raise NotImplementedError
  
      conv = SameLensMaskedConv1d(
          in_feat,
          out_feat,
          kernel_size=kernel_size,
          stride=stride,
          dilation=dilation,
          padding=padding,
          groups=groups,
      )
  
      return conv
  
imports: import math
  import torch
  from torch import nn
  from nemo.collections.tts.modules.submodules import ConvNorm
  from nemo.collections.tts.modules.transformer import PositionalEmbedding
label_desc: Use image node_id 3 for calling a global function with example usage: create_time_mix_layer(in_feat, out_feat, kernel_size, stride, conv_type, dilation) and returns: conv