Home
last modified time | relevance | path

Searched defs:tensor_info_map (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/tests/ut/cpp/stub/parallel_strategy_checkpoint/
Dparallel_strategy_checkpoint_stub.cc32 …tus StrategyCheckpoint::Save(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_map, in Save()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/
Dparallel_strategy_checkpoint.cc142 …tus StrategyCheckpoint::Save(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_map, in Save()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc2657 TensorInfoMap tensor_info_map; in CheckpointStrategy() local