Searched refs:optim_info (Results 1 – 2 of 2) sorted by relevance
33 OptimizerInfo *optim_info = in Build() local35 MS_EXCEPTION_IF_NULL(optim_info); in Build()37 BuildWorkspaces(optim_info, ws_sizes, worker_num); in Build()38 BuildOutputs(optim_info, worker_num); in Build()39 return optim_info; in Build()
291 std::shared_ptr<OptimizerInfo> optim_info = optim_infos_[key]; in UpdateWeights() local292 if (optim_info != nullptr) { in UpdateWeights()293 const std::vector<kernel::AddressPtr> &inputs = optim_info->inputs(); in UpdateWeights()294 const std::vector<kernel::AddressPtr> &workspaces = optim_info->workspaces(); in UpdateWeights()295 const std::vector<kernel::AddressPtr> &outputs = optim_info->outputs(); in UpdateWeights()299 indices_shape.emplace_back(optim_info->indice_size()); in UpdateWeights()310 optim_info->ComputeMean(shapes, worker_num_, pserver_num_, server_node_->rank_id()); in UpdateWeights()312 optim_info->Reset(); in UpdateWeights()327 std::shared_ptr<OptimizerInfo> optim_info = optim_infos_[key]; in AccumGrad() local330 if (optim_info == nullptr) { in AccumGrad()[all …]