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)

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)

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)

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)

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)

This is a sequential container which calls the Conv1d and ReLU modules. During quantization this will be replaced with the corresponding fused module.

ConvReLU2d

class torch.nn.intrinsic.ConvReLU2d(conv, relu)

This is a sequential container which calls the Conv2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.