Searched refs:sparsify_matrix (Results 1 – 9 of 9) sorted by relevance
| /external/libopus/dnn/torch/dnntools/dnntools/sparsification/ |
| D | gru_sparsifier.py | 33 from .common import sparsify_matrix, debug
120 … weight[i * hidden_size : (i+1) * hidden_size, : ], new_mask = sparsify_matrix(
144 … weight[i * hidden_size : (i+1) * hidden_size, : ], new_mask = sparsify_matrix(
|
| D | linear_sparsifier.py | 33 from .common import sparsify_matrix
104 weight[:], new_mask = sparsify_matrix(weight, density, block_size, return_mask=True)
|
| D | conv1d_sparsifier.py | 33 from .common import sparsify_matrix, debug
107 w, new_mask = sparsify_matrix(w, density, block_size, return_mask=True)
|
| D | conv_transpose1d_sparsifier.py | 34 from .common import sparsify_matrix, debug
108 w, new_mask = sparsify_matrix(w, density, block_size, return_mask=True)
|
| D | __init__.py | 5 from .common import sparsify_matrix, calculate_gru_flops_per_step
|
| D | common.py | 34 def sparsify_matrix(matrix : torch.tensor, density : float, block_size, keep_diagonal : bool=False,… function
|
| /external/libopus/dnn/torch/lpcnet/utils/sparsification/ |
| D | gru_sparsifier.py | 32 from .common import sparsify_matrix
135 … gru.weight_ih_l0[i * hidden_size : (i+1) * hidden_size, : ], new_mask = sparsify_matrix(
155 … gru.weight_hh_l0[i * hidden_size : (i+1) * hidden_size, : ], new_mask = sparsify_matrix(
|
| D | __init__.py | 2 from .common import sparsify_matrix, calculate_gru_flops_per_step
|
| D | common.py | 32 def sparsify_matrix(matrix : torch.tensor, density : float, block_size, keep_diagonal : bool=False,… function
|