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.