Home
last modified time | relevance | path

Searched refs:_replicate_for_data_parallel (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/jit/
H A D_script.py578 def _replicate_for_data_parallel(self): member in ScriptModule
579 return self._actual_script_module._replicate_for_data_parallel()
908 def _replicate_for_data_parallel(self): member in RecursiveScriptModule
916 self._c._replicate_for_data_parallel(), init_fn
H A D_script.pyi131 def _replicate_for_data_parallel(self): ...
178 def _replicate_for_data_parallel(self): ...
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dlazy.py285 def _replicate_for_data_parallel(self: _LazyProtocol): member in LazyModuleMixin
H A Drnn.py458 def _replicate_for_data_parallel(self): member in RNNBase
459 replica = super()._replicate_for_data_parallel()
H A Dmodule.py2972 def _replicate_for_data_parallel(self): member in Module
/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A Dreplicate.py160 replica = module._replicate_for_data_parallel()
/aosp_15_r20/external/pytorch/torch/fx/
H A Dgraph_module.py890 def _replicate_for_data_parallel(self): member in GraphModule
/aosp_15_r20/external/pytorch/test/
H A Dtest_fx.py2937 replica = gm._replicate_for_data_parallel()
H A Dtest_nn.py1195 new_param_list = param_list._replicate_for_data_parallel()
1433 new_param_dict = param_dict._replicate_for_data_parallel()