Searched refs:strategy_ckpt_load_file (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | context.h | 116 void set_strategy_ckpt_load_file(const std::string &strategy_ckpt_load_file);
117 std::string strategy_ckpt_load_file() const { return strategy_ckpt_load_file_; } in strategy_ckpt_load_file() function
|
| D | context.cc | 133 void ParallelContext::set_strategy_ckpt_load_file(const std::string &strategy_ckpt_load_file) { in set_strategy_ckpt_load_file() argument
134 strategy_ckpt_load_file_ = strategy_ckpt_load_file; in set_strategy_ckpt_load_file()
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _auto_parallel_context.py | 234 def set_strategy_ckpt_load_file(self, strategy_ckpt_load_file): argument
242 self._context_handle.set_strategy_ckpt_load_file(strategy_ckpt_load_file)
682 parameter_broadcast=bool, strategy_ckpt_load_file=str,
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/ |
| D | parallel_strategy_checkpoint.cc | 33 instance.load_file_ = ParallelContext::GetInstance()->strategy_ckpt_load_file(); in GetInstance()
34 … instance.load_checkpoint_on_ = !ParallelContext::GetInstance()->strategy_ckpt_load_file().empty(); in GetInstance()
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | callbacks.py | 113 strategy_ckpt_load_file=self.config.stra_ckpt)
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_strategy_checkpoint.py | 141 …set_auto_parallel_context(device_num=8, global_rank=0, strategy_ckpt_load_file="./strategy_stage1.…
262 …set_auto_parallel_context(device_num=8, global_rank=0, strategy_ckpt_load_file="./strategy_stage1_…
|
| /third_party/mindspore/mindspore/ |
| D | context.py | 346 … auto_parallel_search_mode=str, parameter_broadcast=bool, strategy_ckpt_load_file=str,
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | init.cc | 175 ….def("get_strategy_ckpt_load_file", &ParallelContext::strategy_ckpt_load_file, "Get strategy check… in PYBIND11_MODULE()
|