Searched refs:_replicate_for_data_parallel (Results 1 – 9 of 9) sorted by relevance
578 def _replicate_for_data_parallel(self): member in ScriptModule579 return self._actual_script_module._replicate_for_data_parallel()908 def _replicate_for_data_parallel(self): member in RecursiveScriptModule916 self._c._replicate_for_data_parallel(), init_fn
131 def _replicate_for_data_parallel(self): ...178 def _replicate_for_data_parallel(self): ...
285 def _replicate_for_data_parallel(self: _LazyProtocol): member in LazyModuleMixin
458 def _replicate_for_data_parallel(self): member in RNNBase459 replica = super()._replicate_for_data_parallel()
2972 def _replicate_for_data_parallel(self): member in Module
160 replica = module._replicate_for_data_parallel()
890 def _replicate_for_data_parallel(self): member in GraphModule
2937 replica = gm._replicate_for_data_parallel()
1195 new_param_list = param_list._replicate_for_data_parallel()1433 new_param_dict = param_dict._replicate_for_data_parallel()