(param_group, master_param)
| 76 | |
| 77 | |
| 78 | def unflatten_master_params(param_group, master_param): |
| 79 | return _unflatten_dense_tensors(master_param, [param for (_, param) in param_group]) |
| 80 | |
| 81 | |
| 82 | def get_param_groups_and_shapes(named_model_params): |
no outgoing calls
no test coverage detected