Searched defs:src_strategy_file (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/ |
| D | checkpoint_transform.py | 76 def rank_list_for_transform(rank_id, src_strategy_file=None, dst_strategy_file=None): argument
139 src_strategy_file=None, dst_strategy_file=None): argument
228 …sform_checkpoint_by_stage(src_checkpoints_dir, dst_checkpoints_dir, ckpt_prefix, src_strategy_file, argument
282 def _transform_checkpoints(src_checkpoints_dir, dst_checkpoints_dir, ckpt_prefix, src_strategy_file… argument
359 def transform_checkpoints(src_checkpoints_dir, dst_checkpoints_dir, ckpt_prefix, src_strategy_file=… argument
|
| D | _parallel_serialization.py | 262 def _extract_src_dst_layout_map_by_src(src_strategy_file=None, dst_strategy_file=None): argument
289 def _extract_src_dst_layout_map(rank_id, src_strategy_file=None, dst_strategy_file=None): argument
|