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�-�@s�
dd
lm
Z

ddlmZmZmZ
ddlZddljZddl m Z mZmZm Z mZmZmZmZ
ddlmZmZmZmZmZmZmZmZmZmZ
Gdd�de
dd d g��Zeeed�Zeeed�Zeeed�Z Gd d�de
dd d g��Z!e!eed�Z"e!ej#ej$d�
ej#ej%d�
d�Z&e!ej#ej%d�
ej#ej%d�
d�Z'e!eed�Z(e!eed�Z)eee d�Z*eejj+e d�Z,eeejj+d�Z-eeed�Z.ddd�
Z/ddd�
Z0eeee!fejj1dd�dd�Z2eee!dfZ3e3eej1dfed�dd�Z4e3e3d�dd�Z5dS) �)
� namedtuple)�Union�Optional�AnyN)�FakeQuantize�default_fake_quant�%default_per_channel_weight_fake_quant�default_weight_fake_quant�default_fused_act_fake_quant�default_fused_wt_fake_quant�FusedMovingAvgObsFakeQuantize�'default_fused_per_channel_wt_fake_quant�
) �HistogramObserver�MovingAverageMinMaxObserver�PlaceholderObserver�default_debug_observer�default_dynamic_quant_observer�default_float_qparams_observer�default_observer�#default_per_channel_weight_observer�default_placeholder_observer�default_weight_observercs eZ
dZd
Z�f
dd�Z�ZS)�QConfiga� Describes how to quantize a layer or a part of the network by providing settings (observer classes) for activations and weights respectively. Note that QConfig needs to contain observer **classes** (like MinMaxObserver) or a callable that returns instances on invocation, not the concrete observer instances themselves. Quantization preparation function will instantiate observers multiple times for each of the layers. Observer classes have usually reasonable default arguments, but they can be overwritten with `with_args` method (that behaves like functools.partial): my_qconfig = QConfig(activation=MinMaxObserver.with_args(dtype=torch.qint8), weight=default_observer.with_args(dtype=torch.qint8)) cs4t|
t
j�st|t
j�r td�
�
tt|�j||
|�S)NzHQConfig received observer instance, please pass observer class instead. zLUse MyObserver.with_args(x=1) to override arguments to constructor if neededz�QConfig received observer instance, please pass observer class instead. Use MyObserver.with_args(x=1) to override arguments to constructor if needed)� isinstance�nn�Module� ValueError�superr�__new__)�cls� activation�weight)
� __class__��I/usr/local/lib64/python3.6/site-packages/torch/ao/quantization/qconfig.pyr/s


zQConfig.__new__)�__name__� __module__�__qualname__�__doc__r� __classcell__r$r$)
r#r%rs
rr!r")r!r")r"r!cs.eZ
dZd
Zejjejjf�f
dd� Z�ZS)�QConfigDynamica� Describes how to dynamically quantize a layer or a part of the network by providing settings (observer classes) for weights. It's like QConfig, but for dynamic quantization. Note that QConfigDynamic needs to contain observer **classes** (like MinMaxObserver) or a callable that returns instances on invocation, not the concrete observer instances themselves. Quantization function will instantiate observers multiple times for each of the layers. Observer classes have usually reasonable default arguments, but they can be overwritten with `with_args` method (that behaves like functools.partial): my_qconfig = QConfigDynamic(weight=default_observer.with_args(dtype=torch.qint8)) cs(t|t
j�rtd�
�
tt|�j||
|�S)NzOQConfigDynamic received observer instance, please pass observer class instead. zLUse MyObserver.with_args(x=1) to override arguments to constructor if neededz�QConfigDynamic received observer instance, please pass observer class instead. Use MyObserver.with_args(x=1) to override arguments to constructor if needed)rrrrrr+r)r r!r")
r#r$r%rPs


zQConfigDynamic.__new__) r&r'r(r)�torchr�Identityrr*r$r$)
r#r%r+@s
r+)
Zdtype�fbgemmc
CsD|d
krtt
jdd�
td�}
n"|dkr�szcadd_module_to_qconfig_obs_ctr..get_factory_kwargs_based_on_module_device..c
Ssh|] }
|
j�qSr$)
rA)rBrCr$r$r%rD�sr
rA) rr,rrr>� parameters�buffers�len�next�iter)ZdevicesrA)
r@r$r%�)get_factory_kwargs_based_on_module_device�s




zPadd_module_to_qconfig_obs_ctr..get_factory_kwargs_based_on_module_devicec
sJy|jdd
�
}
|
�
|j
�d
�
Stk r2


|Stk rD


|SXdS)N)
Zfactory_kwargs)r1Zwith_callable_args�AttributeError� TypeError)Zoriginal_constructor�check)
rJr$r%�1configure_constructor_to_put_obs_on_module_device�s






zXadd_module_to_qconfig_obs_ctr..configure_constructor_to_put_obs_on_module_device)r!r")�_fieldsr!r"rrr+)r4r@rNr!r"r$)rJr@r%�add_module_to_qconfig_obs_ctr�s


rP)�q1�q2cCsvd
d�}|dks|
dkr ||
kS|dk r0|
dk s4t�
y$||j
j|
j
j�oX||jj|
jj�Stk rp


||
kSXdS)NcSs,|j|
jk}|o|j
|
j
k}|o*|j|
jkS)
N)�func�args�keywords)�p1�p2Zsamer$r$r%�partial_equals�s


z&qconfig_equals..partial_equals)r>r!rCr"rK)rQrRrXr$r$r%�qconfig_equals�s


$

rY)
r.)r.r)6�collectionsr�typingrrrr,Ztorch.nnrZ#torch.ao.quantization.fake_quantizerrrr r rrr r7rrrrrrrrrrrr2Zdefault_debug_qconfigZdefault_per_channel_qconfigr+Zdefault_dynamic_qconfigr1Zfloat32Zfloat16Zfloat16_dynamic_qconfigZfloat16_static_qconfigZper_channel_dynamic_qconfigZ!float_qparams_weight_only_qconfigr8r-Zdefault_weight_only_qconfigZdefault_activation_only_qconfigr9r5r;rr?Z QConfigAnyrPrYr$r$r$r%�
sP


(0 















  #


,