Home
last modified time | relevance | path

Searched refs:name_to_attr_creator_ (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel_factory.cc39 (void)name_to_attr_creator_[kernel_name].emplace_back(kernel_attr, kernel_creator); in Register()
53 return (name_to_attr_creator_.find(kernel_name)->second)[ret_pair.second].second(); in Create()
93 auto iter = name_to_attr_creator_.find(kernel_name); in UpdateKernelAttrs()
94 if (iter == name_to_attr_creator_.end()) { in UpdateKernelAttrs()
100 auto creator = name_to_attr_creator_.find(kernel_name)->second[i].second; in UpdateKernelAttrs()
106 auto single_creator = name_to_attr_creator_.find(kernel_name)->second[0].second; in UpdateKernelAttrs()
111 name_to_attr_creator_[kernel_name] = attr_creators; in UpdateKernelAttrs()
116 auto iter = name_to_attr_creator_.find(kernel_name); in CPUKernelAttrCheck()
117 if (iter == name_to_attr_creator_.end()) { in CPUKernelAttrCheck()
167 auto iter = name_to_attr_creator_.find(kernel_name); in GetSupportedKernelAttrList()
[all …]
Dcpu_kernel_factory.h52 std::map<std::string, std::vector<std::pair<KernelAttr, CPUKernelCreator>>> name_to_attr_creator_; variable