View all nncf analysis

How to use the nncf.sparsity.functions.apply_binary_mask function in nncf

To help you get started, we’ve selected a few nncf examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

opencv / openvino_training_extensions / pytorch_toolkit / nncf / nncf / sparsity / layers.py View on Github

def forward(self, weight):
        if is_tracing_state():
            with no_jit_trace():
                return weight.mul_(self.binary_mask)
        tmp_tensor = self._calc_training_binary_mask(weight)
        return apply_binary_mask_impl(tmp_tensor, weight)

opencv / openvino_training_extensions / pytorch_toolkit / nncf / nncf / sparsity / layers.py View on Github

def apply_binary_mask(self, weight):
        return apply_binary_mask_impl(self.binary_mask, weight)