Searched refs:_test_batchnorm_simple_average (Results 1 – 1 of 1) sorted by relevance
11079 def _test_batchnorm_simple_average(self, device, dtype, module_dtype=None): member in TestNNDeviceType11125 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)