Searched defs:_WeightEqualizationObserver (Results 1 – 1 of 1) sorted by relevance
143 class _WeightEqualizationObserver(nn.Module): class210 input_obs: _InputEqualizationObserver, weight_obs: _WeightEqualizationObserver708 weight_eq_obs_dict: Dict[str, _WeightEqualizationObserver], argument