Home
last modified time | relevance | path

Searched refs:ManualShapeMap (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/
Dparallel_strategy_checkpoint.h37 using ManualShapeMap = std::unordered_map<std::string, std::vector<std::pair<int64_t, int64_t>>>; variable
54 …Status Save(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_map, ManualShapeMap
Dparallel_strategy_checkpoint.cc143 ManualShapeMap *manual_shape_map) { in Save()
/third_party/mindspore/tests/ut/cpp/stub/parallel_strategy_checkpoint/
Dparallel_strategy_checkpoint_stub.cc33 ManualShapeMap *manual_shape_map) { return SUCCESS; } in Save()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc2658 ManualShapeMap manual_shape_map; in CheckpointStrategy()