FM / modules

/usr/local/lib64/python3.6/site-packages/torch/nn/intrinsic/qat/modules/

/usr/local/lib64/python3.6/site-packages/torch/nn/intrinsic/qat/modules

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
conv_fused.py	23432	0644	edit dl rm
linear_relu.py	1551	0644	edit dl rm
__init__.py	457	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/nn/intrinsic/qat/modules/__init__.py (457B)

from .linear_relu import LinearReLU from .conv_fused import ( ConvBn1d, ConvBn2d, ConvBn3d, ConvBnReLU1d, ConvBnReLU2d, ConvBnReLU3d, ConvReLU2d, ConvReLU3d, update_bn_stats, freeze_bn_stats, ) __all__ = [ "LinearReLU", "ConvReLU2d", "ConvReLU3d", "ConvBn1d", "ConvBn2d", "ConvBn3d", "ConvBnReLU1d", "ConvBnReLU2d", "ConvBnReLU3d", "update_bn_stats", "freeze_bn_stats", ]