Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/
Dgpu_kernel_factory.cc36 map_kernel_name_to_creater_[kernel_name].emplace_back(kernel_attr, creator); in Register()
57 auto iter = map_kernel_name_to_creater_.find(kernel_name); in SupportedTypeList()
58 if (map_kernel_name_to_creater_.end() == iter) { in SupportedTypeList()
83 auto iter = map_kernel_name_to_creater_.find(kernel_name); in ReducePrecision()
84 if (map_kernel_name_to_creater_.end() == iter) { in ReducePrecision()
128 auto iter = map_kernel_name_to_creater_.find(kernel_name); in GpuKernelAttrCheck()
129 if (map_kernel_name_to_creater_.end() == iter) { in GpuKernelAttrCheck()
187 return (map_kernel_name_to_creater_.find(kernel_name)->second)[ret_pair.second].second(); in Create()
Dgpu_kernel_factory.h64 …map<std::string, std::vector<std::pair<KernelAttr, GpuKernelCreater>>> map_kernel_name_to_creater_; variable