Searched refs:weight_key_to_optims_ (Results 1 – 2 of 2) sorted by relevance
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()88 …<< "Initializing optimizer id for key:" << key << ", optimizer name:" << weight_key_to_optims_[key] in InitWeightKeyToOptims()117 if (weight_key_to_optims_.count(key) > 0) { in InitOptimInputsShape()118 const std::string &optim_name = weight_key_to_optims_[key]; in InitOptimInputsShape()286 if (weight_key_to_optims_.count(key) > 0) { in UpdateWeights()331 … std::shared_ptr<OptimizerInfoBuilder> &builder = optim_info_builders_[weight_key_to_optims_[key]]; in AccumGrad()334 …EXCEPTION) << "no optimizer found for key " << key << " optim name " << weight_key_to_optims_[key]; in AccumGrad()
164 std::unordered_map<Key, std::string> weight_key_to_optims_; variable