Searched refs:InsertWeightInitInfo (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | ps_context.cc | 159 void PSContext::InsertWeightInitInfo(const std::string ¶m_name, size_t global_seed, size_t op_s… in InsertWeightInitInfo() function in mindspore::ps::PSContext
161 ps_cache_instance.InsertWeightInitInfo(param_name, global_seed, op_seed); in InsertWeightInitInfo()
|
| D | ps_context.h | 78 …void InsertWeightInitInfo(const std::string ¶m_name, size_t global_seed, size_t op_seed) const;
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_manager.h | 120 void InsertWeightInitInfo(const std::string ¶m_name, size_t global_seed, size_t op_seed);
|
| D | ps_cache_manager.cc | 67 void PsCacheManager::InsertWeightInitInfo(const std::string ¶m_name, size_t global_seed, size_t… in InsertWeightInitInfo() function in mindspore::ps::PsCacheManager
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | init.cc | 356 ….def("insert_weight_init_info", &PSContext::InsertWeightInitInfo, "Insert embedding table initiali… in PYBIND11_MODULE()
|