Searched defs:module_cls (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | init.py | 7 def skip_init(module_cls, *args, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/api/ |
| H A D | remote_module.py | 92 def _create_module(module_cls, args, kwargs, device): argument
104 module_cls, args, kwargs, device, module_interface_cls argument
|
| /aosp_15_r20/external/pytorch/test/lazy/ |
| H A D | test_extract_compiled_graph.py | 184 def maketest(module_cls, exception_msg_pattern=None, ctxmgr=None): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_modules.py | 63 module_cls = getattr(namespace, module_name) variable
144 def get_module_common_name(module_cls): argument
212 module_cls, # Class object for the module under test argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 8542 def _test_output_unused_in_loss(self, module_cls, gradient_as_bucket_view): argument
|