Searched refs:clone_module_parameter (Results 1 – 3 of 3) sorted by relevance
16 clone_module_parameter,60 sharded_embedding.weight = clone_module_parameter(local_embedding, "weight")
16 clone_module_parameter,67 sharded_embedding_bag.weight = clone_module_parameter(
111 def clone_module_parameter(module, param_name): function