Back to Pytorch

Aliases in torch.ao

docs/source/quantization-support.aliases.md

2.11.05.6 KB
Original Source
{eval-rst}
.. role:: hidden
    :class: hidden-section

Aliases in torch.ao

The following are aliases to their counterparts in torch.ao in nested namespaces.

torch.ao.nn.intrinsic.qat.modules

The following are aliases to their counterparts in torch.ao.nn.intrinsic.qat in the torch.ao.nn.intrinsic.qat.module namespace.

{eval-rst}
.. currentmodule:: torch.ao.nn.intrinsic.qat.modules

torch.ao.nn.intrinsic.qat.modules.conv_fused (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    conv_fused.ConvReLU1d
    conv_fused.ConvReLU2d
    conv_fused.ConvReLU3d
    conv_fused.ConvBnReLU1d
    conv_fused.ConvBnReLU2d
    conv_fused.ConvBnReLU3d

torch.ao.nn.intrinsic.qat.modules.linear_fused (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    linear_fused.LinearBn1d

torch.ao.nn.intrinsic.qat.modules.linear_relu (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    linear_relu.LinearReLU

torch.ao.nn.intrinsic.quantized.modules

{eval-rst}
.. currentmodule:: torch.ao.nn.intrinsic.quantized.modules

The following are aliases to their counterparts in torch.ao.nn.intrinsic.quantized in the torch.ao.nn.intrinsic.quantized.modules namespace.

torch.ao.nn.intrinsic.quantized.modules.conv_relu (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    conv_relu.ConvReLU1d
    conv_relu.ConvReLU2d
    conv_relu.ConvReLU3d

torch.ao.nn.intrinsic.quantized.modules.bn_relu (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    bn_relu.BNReLU2d
    bn_relu.BNReLU3d

torch.ao.nn.intrinsic.quantized.modules.conv_add (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    conv_add.ConvAdd2d
    conv_add.ConvAddReLU2d

torch.ao.nn.intrinsic.quantized.modules.linear_relu (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    linear_relu.LinearLeakyReLU
    linear_relu.LinearReLU
    linear_relu.LinearTanh

torch.ao.nn.intrinsic.quantized.dynamic.modules

{eval-rst}
.. currentmodule:: torch.ao.nn.intrinsic.quantized.dynamic.modules

The following are aliases to their counterparts in the torch.ao.nn.intrinsic.quantized.dynamic namespace.

torch.ao.nn.intrinsic.quantized.dynamic.modules.linear_relu (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    linear_relu.LinearReLU

torch.ao.nn.intrinsic.modules

{eval-rst}
.. currentmodule:: torch.ao.nn.intrinsic.modules

The following are aliases to their counterparts in the torch.ao.nn.intrinsic namespace.

torch.ao.nn.intrinsic.modules.fused (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    fused.ConvAdd2d
    fused.ConvAddReLU2d
    fused.LinearBn1d
    fused.LinearLeakyReLU
    fused.LinearTanh

torch.ao.nn.intrinsic.modules.torch.ao.nn.qat.modules

{eval-rst}
.. currentmodule:: torch.ao.nn.qat.modules

The following are aliases to their counterparts in the torch.ao.nn.qat namespace.

torch.ao.nn.intrinsic.modules.conv (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    conv.Conv1d
    conv.Conv2d
    conv.Conv3d

torch.ao.nn.intrinsic.modules.embedding_ops (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    embedding_ops.Embedding
    embedding_ops.EmbeddingBag

torch.ao.nn.intrinsic.modules.linear (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    linear.Linear

torch.ao.nn.quantizable.modules

{eval-rst}
.. currentmodule:: torch.ao.nn.quantizable.modules

The following are aliases to their counterparts in the torch.ao.nn.quantizable namespace.

torch.ao.nn.quantizable.modules.activation (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    activation.MultiheadAttention

torch.ao.nn.quantizable.modules.rnn (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    rnn.LSTM
    rnn.LSTMCell

torch.ao.nn.quantized.dynamic.modules

{eval-rst}
.. currentmodule:: torch.ao.nn.quantized.dynamic.modules

The following are aliases to their counterparts in the torch.ao.nn.quantized.dynamic namespace.

torch.ao.nn.quantized.dynamic.modules.conv (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    conv.Conv1d
    conv.Conv2d
    conv.Conv3d
    conv.ConvTranspose1d
    conv.ConvTranspose2d
    conv.ConvTranspose3d

torch.ao.nn.quantized.dynamic.modules.linear (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    linear.Linear

torch.ao.nn.quantized.dynamic.modules.rnn (Aliases)

{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    rnn.GRU
    rnn.GRUCell
    rnn.LSTM
    rnn.LSTMCell
    rnn.PackedParameter
    rnn.RNNBase
    rnn.RNNCell
    rnn.RNNCellBase