Searched defs:T_module (Results 1 – 2 of 2) sorted by relevance
80 T_module = TypeVar("T_module", bound=Module) variable83 def weight_norm(module: T_module, name: str = "weight", dim: int = 0) -> T_module:147 def remove_weight_norm(module: T_module, name: str = "weight") -> T_module:
262 T_module = TypeVar("T_module", bound=Module) variable266 module: T_module,337 def remove_spectral_norm(module: T_module, name: str = "weight") -> T_module: