Skip to content

Latest commit

 

History

History
267 lines (208 loc) · 5.62 KB

File metadata and controls

267 lines (208 loc) · 5.62 KB
.. 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.

.. currentmodule:: torch.ao.nn.intrinsic.qat.modules

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

.. 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)

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    linear_fused.LinearBn1d

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

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    linear_relu.LinearReLU

torch.ao.nn.intrinsic.quantized.modules

.. 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)

.. 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)

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    bn_relu.BNReLU2d
    bn_relu.BNReLU3d

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

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    conv_add.ConvAdd2d
    conv_add.ConvAddReLU2d

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

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    linear_relu.LinearLeakyReLU
    linear_relu.LinearReLU
    linear_relu.LinearTanh

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

.. 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)

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    linear_relu.LinearReLU

torch.ao.nn.intrinsic.modules

.. 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)

.. 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

.. 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)

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    conv.Conv1d
    conv.Conv2d
    conv.Conv3d

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

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst


    embedding_ops.Embedding
    embedding_ops.EmbeddingBag

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

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    linear.Linear

torch.ao.nn.quantizable.modules

.. 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)

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    activation.MultiheadAttention

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

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    rnn.LSTM
    rnn.LSTMCell

torch.ao.nn.quantized.dynamic.modules

.. 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)

.. 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)

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

    linear.Linear

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

.. autosummary::
    :toctree: generated
    :nosignatures:
    :template: classtemplate.rst

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