Searched refs:replicate_module_params_buffers (Results 1 – 1 of 1) sorted by relevance
825 def replicate_module_params_buffers(m: nn.Module, mesh: DeviceMesh) -> None: function848 replicate_module_params_buffers(submod, device_mesh)853 replicate_module_params_buffers(submod, device_mesh)