Searched refs:wrapped_with_submodule (Results 1 – 1 of 1) sorted by relevance
121 def wrapped_with_submodule(x: torch.Tensor, batchnorm1d: torch.nn.BatchNorm1d): function485 return wrapped_with_submodule(x, self.batchnorm1d)3097 return wrapped_with_submodule(x, self.batchnorm1d)