/usr/local/lib64/python3.6/site-packages/torch/nn/qat/modules/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/nn/qat/modules/__pycache__/conv.cpython-36.pyc (5141B)
3
ûEg¶ ã @ sJ d dl Z d dljZd dlmZmZ G dd„ dejƒZG dd„ dejƒZdS )é N)Ú
ConvReLU2dÚ
ConvReLU3dc sJ e Zd ZdZejZdddœ‡ fdd „
Zd
d„ Ze dd
„ ƒZ
dd„ Z‡ ZS )ÚConv2daº
A Conv2d module attached with FakeQuantize modules for weight,
used for quantization aware training.
We adopt the same interface as `torch.nn.Conv2d`, please see
https://pytorch.org/docs/stable/nn.html?highlight=conv2d#torch.nn.Conv2d
for documentation.
Similar to `torch.nn.Conv2d`, with FakeQuantize modules initialized to
default.
Attributes:
weight_fake_quant: fake quant module for weight
é r TÚzerosN)Úreturnc
sT ||dœ}
t ƒ j|||f|||||| dœ|
—Ž |
s