diff options
Diffstat (limited to 'dnn/torch/dnntools/dnntools/sparsification/__init__.py')
| -rw-r--r-- | dnn/torch/dnntools/dnntools/sparsification/__init__.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/dnn/torch/dnntools/dnntools/sparsification/__init__.py b/dnn/torch/dnntools/dnntools/sparsification/__init__.py new file mode 100644 index 00000000..fcc91746 --- /dev/null +++ b/dnn/torch/dnntools/dnntools/sparsification/__init__.py @@ -0,0 +1,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
\ No newline at end of file |