Searched refs:fbs_feature_map (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/ |
| D | fused_pull_weight_kernel.h | 178 auto fbs_feature_map = pull_weight_rsp->feature_map(); in ParseFeatureMap() local
179 if (fbs_feature_map->size() != weight_full_names_.size()) { in ParseFeatureMap()
181 << fbs_feature_map->size() << " weights."; in ParseFeatureMap()
185 for (size_t i = 0; i < fbs_feature_map->size(); i++) { in ParseFeatureMap()
186 std::string weight_full_name = fbs_feature_map->Get(i)->weight_fullname()->str(); in ParseFeatureMap()
187 float *weight_data = const_cast<float *>(fbs_feature_map->Get(i)->data()->data()); in ParseFeatureMap()
188 size_t weight_size = fbs_feature_map->Get(i)->data()->size() * sizeof(float); in ParseFeatureMap()
|
| D | update_model_kernel.h | 132 …auto fbs_feature_map = schema::CreateFeatureMap(*(fbb.get()), fbs_weight_fullname, fbs_weight_data… in BuildUpdateModelReq() local
133 fbs_feature_maps.push_back(fbs_feature_map); in BuildUpdateModelReq()
|
| D | fused_push_weight_kernel.h | 153 …auto fbs_feature_map = schema::CreateFeatureMap(*(fbb.get()), fbs_weight_fullname, fbs_weight_data… in BuildPushWeightReq() local
154 fbs_feature_maps.push_back(fbs_feature_map); in BuildPushWeightReq()
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/ |
| D | push_weight_kernel.cc | 125 auto fbs_feature_map = push_weight_req->feature_map(); in ParseFeatureMap() local
126 MS_ERROR_IF_NULL_W_RET_VAL(fbs_feature_map, upload_feature_map); in ParseFeatureMap()
127 for (uint32_t i = 0; i < fbs_feature_map->size(); i++) { in ParseFeatureMap()
128 std::string weight_full_name = fbs_feature_map->Get(i)->weight_fullname()->str(); in ParseFeatureMap()
129 float *weight_data = const_cast<float *>(fbs_feature_map->Get(i)->data()->data()); in ParseFeatureMap()
130 size_t weight_size = fbs_feature_map->Get(i)->data()->size() * sizeof(float); in ParseFeatureMap()
|
| D | update_model_kernel.cc | 232 auto fbs_feature_map = update_model_req->feature_map(); in ParseFeatureMap() local
233 MS_ERROR_IF_NULL_W_RET_VAL(fbs_feature_map, feature_map); in ParseFeatureMap()
234 for (uint32_t i = 0; i < fbs_feature_map->size(); i++) { in ParseFeatureMap()
235 std::string weight_full_name = fbs_feature_map->Get(i)->weight_fullname()->str(); in ParseFeatureMap()
236 float *weight_data = const_cast<float *>(fbs_feature_map->Get(i)->data()->data()); in ParseFeatureMap()
237 size_t weight_size = fbs_feature_map->Get(i)->data()->size() * sizeof(float); in ParseFeatureMap()
|
| D | pull_weight_kernel.cc | 131 …auto fbs_feature_map = schema::CreateFeatureMap(*(fbb.get()), fbs_weight_fullname, fbs_weight_data… in BuildPullWeightRsp() local
132 fbs_feature_maps.push_back(fbs_feature_map); in BuildPullWeightRsp()
|
| D | get_model_kernel.cc | 145 …auto fbs_feature_map = schema::CreateFeatureMap(*(fbb.get()), fbs_weight_fullname, fbs_weight_data… in BuildGetModelRsp() local
146 fbs_feature_maps.push_back(fbs_feature_map); in BuildGetModelRsp()
|
| D | start_fl_job_kernel.cc | 254 …auto fbs_feature_map = schema::CreateFeatureMap(*(fbb.get()), fbs_weight_fullname, fbs_weight_data… in BuildStartFLJobRsp() local
255 fbs_feature_maps.push_back(fbs_feature_map); in BuildStartFLJobRsp()
|