Searched defs:module_dtype (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/test/ |
| D | test_nn.py | 10957 def _test_batchnorm_eval(self, ndim, device, dtype, module_dtype=None): argument
11027 def _test_batchnorm_affine(self, ndim, device, dtype, module_dtype=None): argument
11079 def _test_batchnorm_simple_average(self, device, dtype, module_dtype=None): argument
|