Searched refs:optimizers_ (Results 1 – 2 of 2) sorted by relevance
120 if (optimizers_.count(key) == 0 && optim_inputs_shape_.count(key) > 0) { in InitOptimInputsShape()127 optimizers_[key] = optimizer; in InitOptimInputsShape()132 optimizers_[key] = optimizer; in InitOptimInputsShape()137 optimizers_[key] = optimizer; in InitOptimInputsShape()142 optimizers_[key] = optimizer; in InitOptimInputsShape()287 optimizer = optimizers_[key]; in UpdateWeights()332 std::shared_ptr<kernel::ps::PServerKernel> pserver_kernel = optimizers_[key]; in AccumGrad()
159 std::unordered_map<Key, std::shared_ptr<PServerKernel>> optimizers_; variable