torch.nn.intrinsic
This module implements the combined (fused) modules conv + relu which can be then quantized.
ConvBn1d
-
class
torch.nn.intrinsic.
ConvBn1d
(conv, bn)[source] This is a sequential container which calls the Conv 1d and Batch Norm 1d modules. During quantization this will be replaced with the corresponding fused module.
ConvBn2d
-
class
torch.nn.intrinsic.
ConvBn2d
(conv, bn)[source] This is a sequential container which calls the Conv 2d and Batch Norm 2d modules. During quantization this will be replaced with the corresponding fused module.
ConvBnReLU1d
-
class
torch.nn.intrinsic.
ConvBnReLU1d
(conv, bn, relu)[source] This is a sequential container which calls the Conv 1d, Batch Norm 1d, and ReLU modules. During quantization this will be replaced with the corresponding fused module.
ConvBnReLU2d
-
class
torch.nn.intrinsic.
ConvBnReLU2d
(conv, bn, relu)[source] This is a sequential container which calls the Conv 2d, Batch Norm 2d, and ReLU modules. During quantization this will be replaced with the corresponding fused module.
ConvReLU1d
-
class
torch.nn.intrinsic.
ConvReLU1d
(conv, relu)[source] This is a sequential container which calls the Conv 1d and ReLU modules. During quantization this will be replaced with the corresponding fused module.
ConvReLU2d
-
class
torch.nn.intrinsic.
ConvReLU2d
(conv, relu)[source] This is a sequential container which calls the Conv 2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.