Searched refs:_copy_weights_to_distributed_model (Results 1 – 3 of 3) sorted by relevance
202 dist_utils._copy_weights_to_distributed_model(model, mode)359 dist_utils._copy_weights_to_distributed_model(model, mode)501 dist_utils._copy_weights_to_distributed_model(model, mode)
250 distributed_training_utils_v1._copy_weights_to_distributed_model(
1019 def _copy_weights_to_distributed_model(original_model, mode): function