Searched refs:OPCLASS (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | cpu_kernel_factory.h | 63 #define MS_REG_CPU_KERNEL(OPNAME, ATTR, OPCLASS) MS_REG_CPU_KERNEL_(__COUNTER__, OPNAME, ATTR, OPCL… argument
64 #define MS_REG_CPU_KERNEL_(COUNT, OPNAME, ATTR, OPCLASS) _MS_REG_CPU_KERNEL_(COUNT, OPNAME, ATTR, O… argument
65 #define _MS_REG_CPU_KERNEL_(COUNT, OPNAME, ATTR, OPCLASS) \ argument
66 static_assert(std::is_base_of<CPUKernel, OPCLASS>::value, " must be base of CPUKernel"); \
68 … []() { return std::make_shared<OPCLASS>(); });
70 #define MS_REG_CPU_KERNEL_T(OPNAME, ATTR, OPCLASS, T) MS_REG_CPU_KERNEL_T_(__COUNTER__, OPNAME, ATT… argument
71 …ne MS_REG_CPU_KERNEL_T_(COUNT, OPNAME, ATTR, OPCLASS, T) _MS_REG_CPU_KERNEL_T_(COUNT, OPNAME, ATTR… argument
72 #define _MS_REG_CPU_KERNEL_T_(COUNT, OPNAME, ATTR, OPCLASS, T) \ argument
73 static_assert(std::is_base_of<CPUKernel, OPCLASS<T>>::value, " must be base of CPUKernel"); \
75 #OPNAME, ATTR, []() { return std::make_shared<OPCLASS<T>>(); });
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel_factory.h | 87 #define MS_REG_GPU_KERNEL(OPNAME, OPCLASS) \ argument
88 static_assert(std::is_base_of<GpuKernel, OPCLASS>::value, " must be base of GpuKernel"); \
89 …puKernelRegister UNIQUE_KERNEL_NAME(OPNAME)(#OPNAME, KernelAttr(), []() { return new OPCLASS(); });
92 #define MS_REG_GPU_KERNEL_REGULAR(OPNAME, ATTR, OPCLASS) \ argument
93 static_assert(std::is_base_of<GpuKernel, OPCLASS>::value, " must be base of GpuKernel"); \
94 … const GpuKernelRegister UNIQUE_KERNEL_NAME(OPNAME)(#OPNAME, ATTR, []() { return new OPCLASS(); });
97 #define MS_REG_GPU_KERNEL_SAME(OPNAME, ATTR, OPCLASS, T) \ argument
98 static_assert(std::is_base_of<GpuKernel, OPCLASS<T>>::value, " must be base of GpuKernel"); \
99 …nst GpuKernelRegister UNIQUE_KERNEL_NAME(OPNAME)(#OPNAME, ATTR, []() { return new OPCLASS<T>(); });
102 #define MS_REG_GPU_KERNEL_ONE(OPNAME, ATTR, OPCLASS, T) \ argument
[all …]
|