o
    IñhU  ã                   @   s   d dgZ dd „ Zdd„ ZdS )Ú#get_static_sparse_quantized_mappingÚ$get_dynamic_sparse_quantized_mappingc                  C   s"   dd l } | jj| jjjjji}|S ©Né    )Útorch.ao.nn.sparseÚnnÚLinearÚaoÚsparseÚ	quantized)ÚtorchÚ _static_sparse_quantized_mapping© r   úN/var/www/vscode/kcb/lib/python3.10/site-packages/torch/ao/pruning/_mappings.pyr      s   ÿc                  C   s$   dd l } | jj| jjjjjji}|S r   )r   r   r   r   r	   r
   Údynamic)r   Ú!_dynamic_sparse_quantized_mappingr   r   r   r      s   ÿN)Ú__all__r   r   r   r   r   r   Ú<module>   s
   þ	