Searched defs:_replicate_for_data_parallel (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/nn/modules/ | ||
| H A D | lazy.py | 285 def _replicate_for_data_parallel(self: _LazyProtocol): member in LazyModuleMixin |
| H A D | rnn.py | 458 def _replicate_for_data_parallel(self): member in RNNBase |
| H A D | module.py | 2972 def _replicate_for_data_parallel(self): member in Module |
| /aosp_15_r20/external/pytorch/torch/jit/ | ||
| H A D | _script.py | 578 def _replicate_for_data_parallel(self): member in ScriptModule |
| /aosp_15_r20/external/pytorch/torch/fx/ | ||
| H A D | graph_module.py | 890 def _replicate_for_data_parallel(self): member in GraphModule |