| /third_party/mindspore/mindspore/schema/ |
| D | fl_job.fbs | 67 feature_map:[FeatureMap];
89 feature_map:[FeatureMap];
95 feature_map:[FeatureMap];
105 feature_map:[FeatureMap];
115 feature_map:[FeatureMap];
133 feature_map:[FeatureMap];
145 feature_map:[FeatureMap];
156 feature_map:[FeatureMap];
161 feature_map:[FeatureMap];
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/ |
| D | get_model_kernel.h | 73 auto feature_map = get_model_rsp->feature_map(); in Launch() local
74 MS_EXCEPTION_IF_NULL(feature_map); in Launch()
75 if (feature_map->size() == 0) { in Launch()
79 for (size_t i = 0; i < feature_map->size(); i++) { in Launch()
80 std::string weight_full_name = feature_map->Get(i)->weight_fullname()->str(); in Launch()
81 float *weight_data = const_cast<float *>(feature_map->Get(i)->data()->data()); in Launch()
82 size_t weight_size = feature_map->Get(i)->data()->size() * sizeof(float); in Launch()
|
| D | fused_pull_weight_kernel.h | 109 auto feature_map = ParseFeatureMap(pull_weight_rsp); in Launch() local
112 if (feature_map.count(weight_name) == 0) { in Launch()
116 …memcpy_s(inputs[i]->addr, inputs[i]->size, feature_map[weight_name].addr, feature_map[weight_name]… in Launch()
178 auto fbs_feature_map = pull_weight_rsp->feature_map(); in ParseFeatureMap()
184 std::map<std::string, Address> feature_map; in ParseFeatureMap() local
189 feature_map[weight_full_name] = {weight_data, weight_size}; in ParseFeatureMap()
191 return feature_map; in ParseFeatureMap()
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/ |
| D | update_model_kernel.cc | 188 auto feature_map = ParseFeatureMap(update_model_req); in UpdateModel() local
189 if (feature_map.empty()) { in UpdateModel()
196 for (auto weight : feature_map) { in UpdateModel()
231 std::map<std::string, UploadData> feature_map; in ParseFeatureMap() local
232 auto fbs_feature_map = update_model_req->feature_map(); in ParseFeatureMap()
233 MS_ERROR_IF_NULL_W_RET_VAL(fbs_feature_map, feature_map); in ParseFeatureMap()
241 feature_map[weight_full_name] = upload_data; in ParseFeatureMap()
243 return feature_map; in ParseFeatureMap()
|
| D | pull_weight_kernel.cc | 127 for (auto feature_map : feature_maps) { in BuildPullWeightRsp() local
128 auto fbs_weight_fullname = fbb->CreateString(feature_map.first); in BuildPullWeightRsp()
130 …fbb->CreateVector(reinterpret_cast<float *>(feature_map.second->addr), feature_map.second->size / … in BuildPullWeightRsp()
|
| D | get_model_kernel.cc | 141 for (const auto &feature_map : feature_maps) { in BuildGetModelRsp() local
142 auto fbs_weight_fullname = fbb->CreateString(feature_map.first); in BuildGetModelRsp()
144 …fbb->CreateVector(reinterpret_cast<float *>(feature_map.second->addr), feature_map.second->size / … in BuildGetModelRsp()
|
| D | start_fl_job_kernel.cc | 250 for (auto feature_map : feature_maps) { in BuildStartFLJobRsp() local
251 auto fbs_weight_fullname = fbb->CreateString(feature_map.first); in BuildStartFLJobRsp()
253 …fbb->CreateVector(reinterpret_cast<float *>(feature_map.second->addr), feature_map.second->size / … in BuildStartFLJobRsp()
|
| D | push_weight_kernel.cc | 125 auto fbs_feature_map = push_weight_req->feature_map(); in ParseFeatureMap()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/gnn/ |
| D | graph_loader.cc | 137 std::shared_ptr<Node> *node, NodeFeatureMap *feature_map, in LoadNode() argument
155 (*feature_map)[node_type].insert(ind); in LoadNode()
173 (*feature_map)[node_type].insert(ind); in LoadNode()
186 std::shared_ptr<Edge> *edge, EdgeFeatureMap *feature_map, in LoadEdge() argument
207 (*feature_map)[edge_type].insert(ind); in LoadEdge()
225 (*feature_map)[edge_type].insert(ind); in LoadEdge()
|
| D | graph_loader.h | 87 NodeFeatureMap *feature_map, DefaultNodeFeatureMap *default_feature);
96 EdgeFeatureMap *feature_map, DefaultEdgeFeatureMap *default_feature);
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | executor.h | 73 bool HandleModelUpdateAsync(const std::map<std::string, UploadData> &feature_map);
76 bool HandlePushWeight(const std::map<std::string, Address> &feature_map);
|
| D | executor.cc | 134 bool Executor::HandleModelUpdateAsync(const std::map<std::string, UploadData> &feature_map) { in HandleModelUpdateAsync() argument
136 for (const auto &trainable_param : feature_map) { in HandleModelUpdateAsync()
160 bool Executor::HandlePushWeight(const std::map<std::string, Address> &feature_map) { in HandlePushWeight() argument
161 for (const auto &trainable_param : feature_map) { in HandlePushWeight()
|
| /third_party/mindspore/tests/st/fl/mobile/ |
| D | simulator.py | 91 feature_map = FeatureMap.FeatureMapEnd(builder)
92 feature_maps.append(feature_map)
109 feature_map = builder_update_model.EndVector(len(feature_maps))
115 RequestUpdateModel.RequestUpdateModelAddFeatureMap(builder_update_model, feature_map)
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/ |
| D | simulator.py | 91 feature_map = FeatureMap.FeatureMapEnd(builder)
92 feature_maps.append(feature_map)
109 feature_map = builder_update_model.EndVector(len(feature_maps))
115 RequestUpdateModel.RequestUpdateModelAddFeatureMap(builder_update_model, feature_map)
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | tf_reader_op.cc | 339 …const google::protobuf::Map<std::string, dataengine::Feature> &feature_map = example_features.feat… in LoadExample() local
340 auto iter_column = feature_map.find(current_col.Name()); in LoadExample()
341 if (iter_column == feature_map.end()) { in LoadExample()
585 …const google::protobuf::Map<std::string, dataengine::Feature> &feature_map = example_features.feat… in CreateSchema() local
588 … (void)std::transform(feature_map.begin(), feature_map.end(), std::back_inserter(columns_to_load), in CreateSchema()
594 auto it = feature_map.find(curr_col_name); in CreateSchema()
595 if (it == feature_map.end()) { in CreateSchema()
|
| /third_party/boost/boost/accumulators/framework/ |
| D | depends_on.hpp | 414 feature_map; typedef
419 feature_map
420 , feature_map
|
| /third_party/mindspore/tests/st/networks/models/deeplabv3/src/ |
| D | deeplabv3.py | 319 low_level_feature, feature_map = self.net(x)
322 output = self.aspp(feature_map, scale_index)
330 return feature_map
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | yolo.py | 370 self.feature_map = YOLOv3(backbone=DarkNet(ResidualBlock, self.config.backbone_layers,
383 big_object_output, medium_object_output, small_object_output = self.feature_map(x)
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | thor.py | 428 self.feature_map = Tensor(1.0, mstype.float32)
529 feature_map = self.feature_map
537 feature_map = self.sqrt(1.0 / a_normalizer)
561 matrix_a = self.mul(matrix_a, feature_map)
562 matrix_g = self.mul(matrix_g, feature_map)
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | yolov3.py | 583 self.feature_map = YOLOv3(feature_shape=self.config.feature_shape,
599 big_object_output, medium_object_output, small_object_output = self.feature_map(x)
|
| /third_party/f2fs-tools/include/ |
| D | f2fs_fs.h | 1476 static inline u32 feature_map(struct feature *table, char *feature) in feature_map() function
1486 u32 mask = feature_map(table, features); in set_feature_bits()
|