Home
last modified time | relevance | path

Searched refs:weight_full_names_ (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/
Dfused_pull_weight_kernel.h39 … : server_num_(0), indices_({}), weight_full_names_({}), fl_iteration_(0), total_iteration_(0) {} in FusedPullWeightKernel()
44 if (inputs.size() != weight_full_names_.size()) { in Launch()
46 << weight_full_names_.size() << " weights as inputs."; in Launch()
110 for (size_t i = 0; i < weight_full_names_.size(); i++) { in Launch()
111 const std::string &weight_name = weight_full_names_[i]; in Launch()
121 …MS_LOG(INFO) << "Pull weights for " << weight_full_names_ << " success. Iteration: " << fl_iterati… in Launch()
137 weight_full_names_ = AnfAlgo::GetNodeAttr<std::vector<std::string>>(kernel_node, kAttrPsKey); in Init()
138 … MS_LOG(INFO) << "Weight full name as key " << weight_full_names_ << ", key index is " << indices_ in Init()
140 if (server_num_ == 0 || weight_full_names_.empty() || indices_.empty()) { in Init()
162 for (const std::string &weight_name : weight_full_names_) { in BuildPullWeightReq()
[all …]
Dfused_push_weight_kernel.h37 … : server_num_(0), indices_({}), weight_full_names_({}), fl_iteration_(0), total_iteration_(0) {} in FusedPushWeightKernel()
42 if (inputs.size() != weight_full_names_.size()) { in Launch()
44 << weight_full_names_.size() << " weights as inputs."; in Launch()
109 …MS_LOG(INFO) << "Push weights for " << weight_full_names_ << " success. Iteration: " << fl_iterati… in Launch()
124 weight_full_names_ = AnfAlgo::GetNodeAttr<std::vector<std::string>>(kernel_node, kAttrPsKey); in Init()
125 … MS_LOG(INFO) << "Weight full name as key " << weight_full_names_ << ", key index is " << indices_ in Init()
127 if (server_num_ == 0 || weight_full_names_.empty() || indices_.empty()) { in Init()
148 for (size_t i = 0; i < weight_full_names_.size(); i++) { in BuildPushWeightReq()
149 const std::string &weight_name = weight_full_names_[i]; in BuildPushWeightReq()
168 std::vector<std::string> weight_full_names_; variable
Dupdate_model_kernel.h37 if (inputs.size() != weight_full_names_.size()) { in Launch()
38 …MS_LOG(EXCEPTION) << "Input number of UpdateModelKernel should be " << weight_full_names_.size() <… in Launch()
106 weight_full_names_.push_back(weight_name); in Init()
127 for (size_t i = 0; i < weight_full_names_.size(); i++) { in BuildUpdateModelReq()
128 const std::string &weight_name = weight_full_names_[i]; in BuildUpdateModelReq()
167 std::vector<std::string> weight_full_names_; variable