Searched refs:optimizer_name (Results 1 – 6 of 6) sorted by relevance
49 static std::string optimizer_name(int64_t id);
68 void SetKeyOptimId(size_t key, const std::string &optimizer_name);
59 std::string Util::optimizer_name(int64_t id) { in optimizer_name() function in mindspore::ps::Util
173 void Worker::SetKeyOptimId(size_t key, const std::string &optimizer_name) { in SetKeyOptimId() argument174 MS_LOG(INFO) << "SetKeyOptimId key is:" << key << " optimizer_name:" << optimizer_name; in SetKeyOptimId()175 key_to_optimId_[key] = Util::optimizer_id(optimizer_name); in SetKeyOptimId()
83 if (weight_key_to_optims_.count(key) > 0 || Util::optimizer_name(optim_id) == "") { in InitWeightKeyToOptims()86 weight_key_to_optims_[key] = Util::optimizer_name(optim_id); in InitWeightKeyToOptims()
2672 std::string optimizer_name = AnfAlgo::GetNodeAttr<std::string>(node, kAttrOptimizerType); in AssignParamKey() local2673 ps::Worker::GetInstance().SetKeyOptimId(key, optimizer_name); in AssignParamKey()