Name	Size	Mode	Actions
fake_quantize.cpython-36.pyc	13997	0644	editdlrm
fuser_method_mappings.cpython-36.pyc	4521	0644	editdlrm
fuse_modules.cpython-36.pyc	5012	0644	editdlrm
observer.cpython-36.pyc	39227	0644	editdlrm
qconfig.cpython-36.pyc	8584	0644	editdlrm
quantization_mappings.cpython-36.pyc	6867	0644	editdlrm
quantize.cpython-36.pyc	18901	0644	editdlrm
quantize_jit.cpython-36.pyc	8631	0644	editdlrm
quant_type.cpython-36.pyc	634	0644	editdlrm
stubs.cpython-36.pyc	2716	0644	editdlrm
utils.cpython-36.pyc	6689	0644	editdlrm
_correct_bias.cpython-36.pyc	4449	0644	editdlrm
_equalize.cpython-36.pyc	5704	0644	editdlrm
_learnable_fake_quantize.cpython-36.pyc	6455	0644	editdlrm
__init__.cpython-36.pyc	253	0644	editdlrm

3 Eg?ã@sFdd
lm
Z

Gdd„de
jƒZGdd„de
jƒZGdd„de
jƒZdS) é)
Únncs*eZ
dZd
Zd‡f
dd„ Zdd„Z‡ZS)Ú QuantStuba
Quantize stub module, before calibration, this is same as an observer, it will be swapped as `nnq.Quantize` in `convert`. Args: qconfig: quantization configuration for the tensor, if qconfig is not provided, we will get qconfig from parent modules Ncstt
|ƒjƒ
|
r|
|_dS)
N)ÚsuperrÚ__init__Úqconfig)Úselfr)
Ú __class__©úG/usr/local/lib64/python3.6/site-packages/torch/ao/quantization/stubs.pyrs


zQuantStub.__init__c
Cs|
S)
Nr )rÚ
xr r r Úforwards
zQuantStub.forward)
N)Ú__name__Ú __module__Ú__qualname__Ú__doc__rrÚ __classcell__r r )
rr rs
rcs(eZ
dZd
Z‡f
dd„Zdd„Z‡ZS)ÚDeQuantStubz…Dequantize stub module, before calibration, this is same as identity, this will be swapped as `nnq.DeQuantize` in `convert`. c

stt
|ƒjƒ
dS)
N)rrr)
r)
rr r rs
zDeQuantStub.__init__c
Cs|
S)
Nr )rrr r r rs
zDeQuantStub.forward)r rrrrrrr r )
rr rs
rcs8eZ
dZUd
Zeeej  ‡f
dd„Zdd„Z‡Z S)ÚQuantWrapperaç
A wrapper class that wraps the input module, adds QuantStub and DeQuantStub and surround the call to module with call to quant and dequant modules. This is used by the `quantization` utility functions to add the quant and dequant modules, before `convert` function `QuantStub` will just be observer, it observes the input tensor, after `convert`, `QuantStub` will be swapped to `nnq.Quantize` which does actual quantization. Similarly for `DeQuantStub`. cs\tt
|ƒjƒ
t|
d
ƒr|
jnd}|jdt|ƒ
ƒ
|jdtƒƒ
|jd|
ƒ
|j|
j ƒ

dS)NrÚquantÚdequantÚmodule) rrrÚhasattrrZ add_modulerrZtrainZtraining)rrr)
rr r r/s





zQuantWrapper.__init__cCs|j|
ƒ
}
|j
|
ƒ
}
|j|
ƒ
S)
N)rrr)rÚ
Xr r r r7s


zQuantWrapper.forward)r rrrrrrrrÚModulerrrrr r )
rr r s 


rN)Ztorchrrrrrr r r r Ús