FM / __pycache_

/usr/local/lib64/python3.6/site-packages/torch/ao/ns/fx/__pycache__/

/usr/local/lib64/python3.6/site-packages/torch/ao/ns/fx/__pycache__

Name	Size	Mode	Actions
graph_matcher.cpython-36.pyc	12023	0644	edit dl rm
graph_passes.cpython-36.pyc	15354	0644	edit dl rm
mappings.cpython-36.pyc	7209	0644	edit dl rm
ns_types.cpython-36.pyc	809	0644	edit dl rm
pattern_utils.cpython-36.pyc	3795	0644	edit dl rm
utils.cpython-36.pyc	10942	0644	edit dl rm
weight_utils.cpython-36.pyc	6573	0644	edit dl rm
__init__.cpython-36.pyc	150	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/ao/ns/fx/__pycache__/mappings.cpython-36.pyc (7209B)

3 Eg�9�@sHddlZddlZddljZddljjZejjZ ddl jjZddljjj ZddljjjZddljjjj ZddljjjZddljjZddljjZddlmZddlmZmZm Z m!Z!ee"eefd�dd�Z#ee"eefee!e"d�d d �Z$ee"eefee!edd�dd �Z%ee"eefd�dd�Z&ee"eefd�dd�Z'dS)�N�)�NSNodeTargetType)�Set�Dict�List�Optional)�returncCCs>ttjtjtjtjtjt jg�ttj tj tj tjtj tjtjt jg�ttjtjtjtjtjtjtjt jg�ttjtjtjg�ttjtjtjg�ttjtjtjg�ttjtjt jtjtjtjtjtjtjj j!g �ttj tj tj"g�ttj#t$j%g�ttj&t$j'j(j)g�ttj*t$j'j(j+g�ttj,tj-g�ttj.tj/g�ttj0tj1g�ttj2tj2g�tt$j3tj3t4j3tj5g�tt$j6tj6g�tt$j7tj7t4j7tj8g�ttj9tj:ddt$j9g�ttj;tjg�ttj?tj@g�tt$jAddtjBtjAg�ttjCtjCg�ttjDtjDg�ttjEtjEg�ttjFtjFg�ttjGtjGg�ttjHtjHg�ttjItjIg�ttjJtjJg�ttjKtjKg�ttjLtjLg�ttjMtjMg�ttjNtjNg�ttjOtjOg�ttjPtjPg�ttjQtjRtjQg�tt jStjSg�tt jTtjTg�ttjUtjUg�ttjVtjVg�ttjWtjWg�ttjXtjXg�ttjYtjYg�ttjZtj[g�ttj\tj]g�ttj^tj_t$j_ddg�tddtj`tjag�ttjbtjctjdg�tt4jeg�tt$jfg�tt$jgg�tt$jhg�tt$jig�tt$jjg�tt$jkg�tt$jlg�tt$jmg�tt$jng�tt$jog�tt$jpg�tt$jqg�tt$jjrjsg�ttjttjug�gA}i}d }x&|D]}tv|�}|d 7}|||<�qW|S)N�relu�relu_�sigmoid�sigmoid_�tanh_�tanh�hardsigmoid_�hardsigmoidrr)w�set�nn�Conv1d�nnq�nniqat�ConvBn1d�ConvBnReLU1d�nniq� ConvReLU1d�nni�Conv2d�nnqat�ConvBn2d�ConvBnReLU2d� ConvReLU2d�Conv3d�ConvBn3d�ConvBnReLU3d� ConvReLU3d�F�conv1d�toq�conv1d_relu�conv2d�conv2d_relu�conv3d�conv3d_relu�Linear� LinearReLU�nniqd�nnqd�modules�linear�NonDynamicallyQuantizableLinear�linear_relu� AvgPool1d�torch� avg_pool1d� AvgPool2d�_C�_nn� avg_pool2d� AvgPool3d� avg_pool3d�AdaptiveAvgPool1d�adaptive_avg_pool1d�AdaptiveAvgPool2d�adaptive_avg_pool2d�AdaptiveAvgPool3d�adaptive_avg_pool3d�LSTM�add�operatorZadd_relu�cat�mulZmul_relur �ReLU� MaxPool1d� max_pool1d� MaxPool2d� max_pool2d� MaxPool3d� max_pool3dr�Sigmoid�BatchNorm2d�BatchNorm3d�ConvTranspose1d�ConvTranspose2d�ELUZ EmbeddingZEmbeddingBag� GroupNorm� Hardswish�InstanceNorm1d�InstanceNorm2d�InstanceNorm3d� LayerNorm� LeakyReLU�ReLU6�relu6�BNReLU2d�BNReLU3d�elu� hardswish� instance_norm� layer_norm� leaky_relu�SiLU�silu�Mish�mish�Tanhrr�Hardsigmoid�Hardtanh�hardtanh� hardtanh_�floordiv� unsqueeze�stack�squeeze�sort�repeat_interleave�min�mean�max� transpose�flatten�clamp�chunk� functional�interpolate�Dropout�dropout�str)Zsets_of_related_ops� base_name_to_sets_of_related_ops�counter�set_of_related_ops� base_name�r��C/usr/local/lib64/python3.6/site-packages/torch/ao/ns/fx/mappings.py�$get_base_name_to_sets_of_related_opss� r�)r��oprcCs&x |j�D]\}}||kr |Sq WdS)N)�items)r�r�r�r�r�r�r��get_base_name_for_opxsr�)r�r�� related_oprcCsx|dk rDx*|j�D]\}}||kr|j|�dSqWt|�d��n0d}xt|�|kr`|d7}qJWt|g�|t|�<dS)Nz was not foundrr)r�rD�AssertionErrorrr)r�r�r�r�r�r�r�r�r��add_op_to_sets_of_related_ops�s r�c.Cs�ttjtjtjtjtjtjtj tj tjtjtj tjtjg �}t�}ttjtjtjtjtjtjtjtjtjtjtj tj tjtjg�}ttjtjtjtjtjtjtjtjtjtjtjtjtj tj!tj"tj#tj$tj%tj&tj'j(j)tj'j(j*tjtj+tj,tj-tj.tj/tj0tj1tj2tj3tj4tj5tj6g"�}tt7j8t9j8t:j8tj7j;jjt7j?t9j>t9j?t7j@t:j@t7jAt7jBt7jCt7jDt7jEt7jFt7jGt7jHt7jIt7jJt7jKt7jLt7jMt7jNt7jOtPjQtPjRtPjStPjTtPjUtPjVtPjWtPjXtPjYtZjWtZjXtZjYtZj[tZj\tZj]tZjTtZjUtZjVt^jVg-�}tt_j8t_j=t_j>t`jTt_j?t_jAt_jBt_jCt_jDt_jEt_jFt_jGt_jHt_jIt_jJt_jKt_jLt_jMt`jQt`jRt`jSt`jTt`jUt`jVg�}tt7jat7jbt7jct7jdt7jet7jft7jgt7jht7jit7jjt7jkt7jlt7jmt7jnt7jot7jpt7jMg�}tddddddddg�}||||||||d �S) Nrrr rrrr r )Zfuns_io_type_fp32Zfuns_io_type_fp16Zfuns_io_type_int8Zfuns_io_type_fp32_or_int8Zmods_io_type_fp32Zmods_io_type_int8Zmods_io_type_fp32_or_int8Zmeths_io_type_fp32_or_int8)qrr$r1r%r(r*r5rFr`rarbrcrdrfrh�sumr&r3r'r)r+r rrrrErnr>r@rBr~rlrmr|rJrLrNr]r6r8r9r:r<rzryrxrwrvrurtrsrrrqrprorr,rr/r0r2rrr rCrPrQrRrSrTrUrWrXrYrZrVr[r\rergrr^r_rrr#r-rrr!rrrr"r.rrrHrirOrjr=r?rAr4r7r;r}rk�IdentityrIrKrM)ZFUNS_IO_TYPE_FP32ZFUNS_IO_TYPE_FP16ZFUNS_IO_TYPE_INT8ZFUNS_IO_TYPE_FP32_OR_INT8ZMODS_IO_TYPE_FP32ZMODS_IO_TYPE_INT8ZMODS_IO_TYPE_FP32_OR_INT8ZMETHS_IO_TYPE_FP32_OR_INT8r�r�r��get_node_type_to_io_type_map�sV r�cCs\ttjtjg�}ttjg�}tddddddddd d ddd dddddddddg�}|||d�S)NZtoZ dequantizeZreshape�viewZ unsqueeze_rorwZsqueeze_rq�size�shapeZresize_rs�repeatZpermuteZnumelruZdetach_�detach� contiguousryrz)Zfuns_unmatchableZmods_unmatchableZmeths_unmatchable)rr5Zquantize_per_tensorrE�getitemrr�)ZFUNS_UNMATCHABLEZMODS_UNMATCHABLEZMETHS_UNMATCHABLEr�r�r��get_unmatchable_types_map_s> r�)(rEr5Ztorch.nnrZtorch.nn.functionalr{r$�opsZ quantizedr&Ztorch.nn.quantizedrZtorch.nn.quantized.dynamicZdynamicr/Ztorch.nn.intrinsic.quantizedZ intrinsicrZ$torch.nn.intrinsic.quantized.dynamicr.Ztorch.nn.intrinsic.qatZqatrZtorch.nn.intrinsicrZtorch.nn.qatrZns_typesr�typingrrrrrr�r�r�r�r�r�r�r�r��s4 fJ