Home
last modified time | relevance | path

Searched refs:_remove_repeated_slices (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/parallel/
D_utils.py300 def _remove_repeated_slices(tensor_layout): function
324 new_train_layout = _remove_repeated_slices(train_layout)
/third_party/mindspore/mindspore/train/
Dserialization.py47 from mindspore.parallel._utils import _infer_rank_list, _remove_repeated_slices
1408 param_unique_strategy = _remove_repeated_slices(train_strategy[param.name])