Searched refs:_update_mask (Results 1 – 1 of 1) sorted by relevance
373 def _update_mask(self, weights, threshold): member in Pruning446 return self._update_mask(weights, threshold)477 smoothed_threshold, new_mask = self._update_mask(pooled_weights,