Searched refs:module_contains_param (Results 1 – 4 of 4) sorted by relevance
14 from .parametrization import BiasHook, FakeStructuredSparsity, module_contains_param296 and module_contains_param(first_module, FakeStructuredSparsity)307 if module_contains_param(module, FakeStructuredSparsity):
7 def module_contains_param(module, parametrization): function
15 module_contains_param,324 module_contains_param(mod, parameterization)
19 def module_contains_param(module: nn.Module, parametrization: Type[nn.Module]) -> bool: function