o
    Iñh€  ã                   @   st   d dl mZmZ d dlmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ d dlmZmZmZmZ d dlmZ d	S )
é   )Ú$get_dynamic_sparse_quantized_mappingÚ#get_static_sparse_quantized_mapping)ÚBaseScheduler)ÚCubicSL)ÚLambdaSL)ÚBaseSparsifier)ÚNearlyDiagonalSparsifier)ÚFakeSparsityÚfqn_to_moduleÚget_arg_info_from_tensor_fqnÚmodule_to_fqn)ÚWeightNormSparsifierN)Ú	_mappingsr   r   Úscheduler.base_schedulerr   Úscheduler.cubic_schedulerr   Úscheduler.lambda_schedulerr   Úsparsifier.base_sparsifierr   Ú%sparsifier.nearly_diagonal_sparsifierr   Úsparsifier.utilsr	   r
   r   r   Ú!sparsifier.weight_norm_sparsifierr   © r   r   úM/var/www/vscode/kcb/lib/python3.10/site-packages/torch/ao/pruning/__init__.pyÚ<module>   s   