Searched refs:_create_module (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/distributed/nn/api/ |
| D | remote_module.py | 92 def _create_module(module_cls, args, kwargs, device): function
106 module = _create_module(module_cls, args, kwargs, device)
278 _create_module,
|
| /external/pytorch/test/distributed/fsdp/ |
| D | test_fsdp_state_dict.py | 942 def _create_module(wrap_fsdp=True): function
963 fsdp, linear_skip_tensor_names = _create_module()
980 new_fsdp, _ = _create_module()
993 local, _ = _create_module(wrap_fsdp=False)
|
| /external/pigweed/pw_module/py/pw_module/ |
| D | create.py | 1082 def _create_module( function
1227 parser.set_defaults(func=_create_module)
|