Searched defs:lazy_cls (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_lazy_modules.py | 171 lazy_cls, argument
197 lazy_cls, argument
505 def _check_lazy_norm(self, cls, lazy_cls, input_shape): argument
556 def _check_lazy_norm_pickle(self, cls, lazy_cls, input_shape): argument
585 def _check_lazy_batchnorm_state(self, cls, lazy_cls): argument
603 def _check_lazy_instancenorm_state(self, cls, lazy_cls): argument
627 def _check_lazy_norm_with_dict_input(self, cls, lazy_cls, input_shape): argument
|