Searched refs:_maybe_update_block_mask (Results 1 – 2 of 2) sorted by relevance
422 def _maybe_update_block_mask(self, weights, threshold): member in Pruning511 new_threshold, new_mask = self._maybe_update_block_mask(weight, threshold)
127 _, new_mask = p._maybe_update_block_mask(weights, threshold)