dmx.compressor.modeling.nn.torch\_modules
=========================================

.. automodule:: dmx.compressor.modeling.nn.torch_modules
   :members:
   :undoc-members:
   :show-inheritance:

   
   .. rubric:: Classes

   .. autosummary::
      :toctree:
   
      ActActMatMul
      AdaptiveAvgPool2d
      AvgPool2d
      BAddBMM
      BatchNorm2d
      Conv1d
      Conv2d
      ConvTranspose2d
      Dropout
      Embedding
      Exp
      GELU
      GELUBase
      GroupNorm
      LayerNorm
      Linear
      MaxPool2d
      Mul
      RMSNorm
      ReLU
      ReLU6
      ResAdd
      ScaledDotProductAttention
      SiLU
      Softmax
      Tanh