torch.nn.intrinsic.quantized | |
-------------------------------------- | |
This module implements the quantized implementations of fused operations like conv + relu. | |
.. automodule:: torch.nn.intrinsic.quantized | |
ConvReLU2d | |
~~~~~~~~~~~~~~~ | |
.. autoclass:: ConvReLU2d | |
:members: | |
ConvReLU3d | |
~~~~~~~~~~~~~~~ | |
.. autoclass:: ConvReLU3d | |
:members: | |
LinearReLU | |
~~~~~~~~~~~~~~~ | |
.. autoclass:: LinearReLU | |
:members: | |