blob: fcc91746776576bf35326cf75111380bafeaa657 (
plain)
1
2
3
4
5
6
|
from .gru_sparsifier import GRUSparsifier
from .conv1d_sparsifier import Conv1dSparsifier
from .conv_transpose1d_sparsifier import ConvTranspose1dSparsifier
from .linear_sparsifier import LinearSparsifier
from .common import sparsify_matrix, calculate_gru_flops_per_step
from .utils import mark_for_sparsification, create_sparsifier
|
