Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py11079 def _test_batchnorm_simple_average(self, device, dtype, module_dtype=None): member in TestNNDeviceType
11125 self._test_batchnorm_simple_average(device, dtype)
11129 self._test_batchnorm_simple_average(device, dtype)
11134 self._test_batchnorm_simple_average(device, dtype, torch.float)
11138 self._test_batchnorm_simple_average(device, dtype, torch.float)