Searched defs: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
70 #define MS_REG_CPU_KERNEL_T(OPNAME, ATTR, OPCLASS, T) MS_REG_CPU_KERNEL_T_(__COUNTER__, OPNAME, ATT… argument
71 #define MS_REG_CPU_KERNEL_T_(COUNT, OPNAME, ATTR, OPCLASS, T) _MS_REG_CPU_KERNEL_T_(COUNT, OPNAME, … argument
72 #define _MS_REG_CPU_KERNEL_T_(COUNT, OPNAME, ATTR, OPCLASS, T) \ argument
77 #define MS_REG_CPU_KERNEL_T_S(OPNAME, ATTR, OPCLASS, T, S) \ argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel_factory.h | 87 #define MS_REG_GPU_KERNEL(OPNAME, OPCLASS) \ argument
92 #define MS_REG_GPU_KERNEL_REGULAR(OPNAME, ATTR, OPCLASS) \ argument
97 #define MS_REG_GPU_KERNEL_SAME(OPNAME, ATTR, OPCLASS, T) \ argument
102 #define MS_REG_GPU_KERNEL_ONE(OPNAME, ATTR, OPCLASS, T) \ argument
107 #define MS_REG_GPU_KERNEL_TWO(OPNAME, ATTR, OPCLASS, T, S) \ argument
112 #define MS_REG_GPU_KERNEL_THREE(OPNAME, ATTR, OPCLASS, T, S, G) … argument
|