Searched refs:_copy_weights_to_distributed_model (Results 1 – 3 of 3) sorted by relevance
197 dist_utils._copy_weights_to_distributed_model(model, mode)354 dist_utils._copy_weights_to_distributed_model(model, mode)496 dist_utils._copy_weights_to_distributed_model(model, mode)
246 distributed_training_utils_v1._copy_weights_to_distributed_model(
1009 def _copy_weights_to_distributed_model(original_model, mode): function