Home
last modified time | relevance | path

Searched refs:wrapped_with_submodule (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_fx.py121 def wrapped_with_submodule(x: torch.Tensor, batchnorm1d: torch.nn.BatchNorm1d): function
485 return wrapped_with_submodule(x, self.batchnorm1d)
3097 return wrapped_with_submodule(x, self.batchnorm1d)