Searched defs:tensor_info_map (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/ |
| D | strategy_checkpoint_info.h | 44 void Init(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_map, in Init()
53 TensorInfoMap tensor_info_map() const { return tensor_info_map_; } in tensor_info_map() function
|
| D | parallel_strategy_checkpoint.cc | 139 …tus StrategyCheckpoint::Save(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_map, in Save()
214 …eckpoint::SaveAutoOpStrategy(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_map, in SaveAutoOpStrategy()
|
| D | strategy_checkpoint_info.cc | 26 void StrategyCheckpointInfo::set_tensor_info_map(const TensorInfoMap &tensor_info_map) { in set_tensor_info_map()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/stub/parallel_strategy_checkpoint/ |
| D | parallel_strategy_checkpoint_stub.cc | 32 …tus StrategyCheckpoint::Save(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_map, in Save()
42 …eckpoint::SaveAutoOpStrategy(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_map, in SaveAutoOpStrategy()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_auto_parallel.cc | 1612 TensorInfoMap tensor_info_map; in SaveStrategyToFile() local
|
| D | step_parallel.cc | 2820 TensorInfoMap tensor_info_map; in CheckpointStrategy() local
|