Searched refs:GpuKernelCreater (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel_factory.h | 32 using GpuKernelCreater = std::function<GpuKernel *()>; variable
39 …void Register(const std::string &kernel_name, const KernelAttr &kernel_attr, GpuKernelCreater &&cr…
62 … std::vector<std::pair<KernelAttr, GpuKernelCreater>> *iter_second, size_t attr_index);
64 …std::map<std::string, std::vector<std::pair<KernelAttr, GpuKernelCreater>>> map_kernel_name_to_cre…
69 …GpuKernelRegister(const std::string &kernel_name, const KernelAttr &kernel_attr, GpuKernelCreater … in GpuKernelRegister()
|
| D | gpu_kernel_factory.cc | 35 GpuKernelCreater &&creator) { in Register()
40 … std::vector<std::pair<KernelAttr, GpuKernelCreater>> *iter_second, in CheckIOParam()
|