Searched refs:InitWeightKeyToOptims (Results 1 – 2 of 2) sorted by relevance
123 void InitWeightKeyToOptims(const Key &key, const int64_t &optim_id);
82 void ParameterServer::InitWeightKeyToOptims(const Key &key, const int64_t &optim_id) { in InitWeightKeyToOptims() function in mindspore::ps::ParameterServer695 ps_->InitWeightKeyToOptims(key, static_cast<int64_t>(val)); in HandleInitWeightToOptimId()