Searched refs:InsertAccumuInitInfo (Results 1 – 5 of 5) sorted by relevance
165 void PSContext::InsertAccumuInitInfo(const std::string ¶m_name, float init_val) const { in InsertAccumuInitInfo() function in mindspore::ps::PSContext167 ps_cache_instance.InsertAccumuInitInfo(param_name, init_val); in InsertAccumuInitInfo()
79 void InsertAccumuInitInfo(const std::string ¶m_name, float init_val) const;
121 void InsertAccumuInitInfo(const std::string ¶m_name, float init_val);
88 void PsCacheManager::InsertAccumuInitInfo(const std::string ¶m_name, float init_val) { in InsertAccumuInitInfo() function in mindspore::ps::PsCacheManager
357 ….def("insert_accumu_init_info", &PSContext::InsertAccumuInitInfo, "Insert accumulation initializat… in PYBIND11_MODULE()