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�dZd
dl
Z
d
dlZd
dlZd
dlmZmZ
d
dlmZm Z
dd�Z dd�Zd d �Ze e e d�dd �Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zd d!�Zejejed"�d#d$�Zeeeejeeeefd%�d&d'�ZdS)(z? Utils shared by different modes of quantization (eager/graph) �N)� QuantType�quant_type_to_str)�Tuple�AnycCs|j�}|j
|
�

|S)
N)�copy�update)Zdefault_dictZadditional_dict�
d�r �G/usr/local/lib64/python3.6/site-packages/torch/ao/quantization/utils.py�get_combined_dict s


rc

Cs|tj
kp|tjkS)
N)�torch�per_tensor_affineZper_tensor_symmetric)
�qschemer r r � is_per_tensors

rc

Cs|tj
tjtjgkS)
N)r�per_channel_affineZ per_channel_affine_float_qparams�per_channel_symmetric)
rr r r �is_per_channels


r)�obj�fqn�returncCstj
t|
jd
�
|�S)zO Given an obj and a fqn such as "foo.bar.baz", returns gm.foo.bar.baz. �
.)� functools�reduce�getattr�split)rrr r r �getattr_from_fqnsrc
Cs�t|d
�r|j
nd}
|j}|
|d�}|
s,|St|
�
r|d
k st�
|j
�}
|j�}t|
d�r,|
jnd
}|
j|j|fS)zk returns the qconfig tuple for qconfig: (activation_dtype, weight_dtype, activation_compute_dtype) N� compute_dtype)r%r)r+rr4r)r'r)r+r4r r r �get_qconfig_dtypesos 



r5c
Cs�|dk st�
|j
�}
|j�}tjtjg}|j|krb|
j|krBtjSt |
d
�r\|
j |kr\tjStjS|jtj kr�|
jtjkr�tjS|
jtj kr�tjStdj|
j|j��
�
dS)Nr4zKUnrecognized dtype combination in get_quant_type: activation({}),weight({}))r%r)r+rr-r.rrZSTATICrr4ZDYNAMICZWEIGHT_ONLYr/�float� Exceptionr&)r'r)r+Z static_dtypesr r r r"ys"













r")�min_val�max_valrcCs�|j�d
ks|
j�d
kr&t
jd �

dS|j�d
ks>|
j�d
kr~|td�
krd|
td�
krdt
jd �

dS||
k
s�tdj||
��
�
ntj||
k
�
s�tdj||
��
�
dS)z� Checks if the given minimum and maximum values are valid, meaning that they exist and the min value is less than the max value. r
z4must run observer before calling calculate_qparams. zReturning default values.F�infz-infz!min {} should be less than max {}TzMmust run observer before calling calculate_qparams. Returning default values.zMmust run observer before calling calculate_qparams. Returning default values.) Znumel�warnings�warnZdimr6r%r&r�all)r8r9r r r �check_min_max_valid�s"









r>)� quant_min� quant_max�has_customized_qranger�reduce_rangerc Cs�|r�d \}}||
}}|dk r0|dk r0||}}||d} d
| koNdk
n
s\td�
�
|t
jkr�| d| dd}}
nd
| d}}
|r�|d|
d}}
nP|t
jkr�|r�d\}}
nd\}}
n.|t
jk�r�|�r�d\}}
nd\}}
nd\}}
||
fS)ztCalculates actual qmin and qmax based on the quantization range, observer datatype and if range is reduced. r
��N�
�
zRquantization range should be positive and not exceed the maximum bit range (=256).��@�?���)r
rC�����)rLrH����)rMrJ)r
rJ)r
rC)r
rK)r%rr.r-) r?r@rArrBZinitial_quant_minZinitial_quant_maxZcustom_quant_minZcustom_quant_maxZ qrange_lenr r r �calculate_qmin_qmax�s0











 
rN)�__doc__r;rrZ torch.ao.quantization.quant_typerr�typingrrrrr�strrr!r(r*r,r0r1r2r3r5r"ZTensor�boolr>�intrrNr r r r �s,