Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Dkernel_factory.h57 name_to_creator_map_[name].push_back(std::make_pair(params_info, creator)); in Register()
62 if (name_to_creator_map_.count(name) == 0) { in Create()
65 for (const auto &name_type_creator : name_to_creator_map_[name]) { in Create()
86 std::unordered_map<std::string, std::vector<std::pair<ParamsInfo, C>>> name_to_creator_map_; variable
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/
Dround_kernel_factory.cc29 name_to_creator_map_[name] = creator; in Register()
33 if (name_to_creator_map_.count(name) == 0) { in Create()
37 auto kernel = name_to_creator_map_[name](); in Create()
Dround_kernel_factory.h45 std::unordered_map<std::string, RoundKernelCreator> name_to_creator_map_; variable