Home
last modified time | relevance | path

Searched refs:MS_REG_CPU_KERNEL_T_S (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcast_cpu_kernel.h45 MS_REG_CPU_KERNEL_T_S(Cast, KernelAttr(), CastCPUKernel, uint8_t, uint8_t);
46 MS_REG_CPU_KERNEL_T_S(Cast, KernelAttr(), CastCPUKernel, uint8_t, uint16_t);
47 MS_REG_CPU_KERNEL_T_S(Cast, KernelAttr(), CastCPUKernel, uint8_t, uint32_t);
48 MS_REG_CPU_KERNEL_T_S(Cast, KernelAttr(), CastCPUKernel, uint8_t, uint64_t);
49 MS_REG_CPU_KERNEL_T_S(Cast, KernelAttr(), CastCPUKernel, uint8_t, int8_t);
50 MS_REG_CPU_KERNEL_T_S(Cast, KernelAttr(), CastCPUKernel, uint8_t, int16_t);
51 MS_REG_CPU_KERNEL_T_S(Cast, KernelAttr(), CastCPUKernel, uint8_t, int32_t);
52 MS_REG_CPU_KERNEL_T_S(Cast, KernelAttr(), CastCPUKernel, uint8_t, int64_t);
53 MS_REG_CPU_KERNEL_T_S(Cast, KernelAttr(), CastCPUKernel, uint8_t, float16);
54 MS_REG_CPU_KERNEL_T_S(Cast, KernelAttr(), CastCPUKernel, uint8_t, float);
[all …]
Dscatter_nd_cpu_kernel.h58 MS_REG_CPU_KERNEL_T_S(
63 MS_REG_CPU_KERNEL_T_S(
68 MS_REG_CPU_KERNEL_T_S(
72 MS_REG_CPU_KERNEL_T_S(
76 MS_REG_CPU_KERNEL_T_S(
80 MS_REG_CPU_KERNEL_T_S(
84 MS_REG_CPU_KERNEL_T_S(
89 MS_REG_CPU_KERNEL_T_S(
94 MS_REG_CPU_KERNEL_T_S(
99 MS_REG_CPU_KERNEL_T_S(
[all …]
Dgather_d_grad_cpu_kernel.h45 MS_REG_CPU_KERNEL_T_S(GatherDGrad, KernelAttr(), GatherDGradCPUKernel, int32_t, int32_t);
46 MS_REG_CPU_KERNEL_T_S(GatherDGrad, KernelAttr(), GatherDGradCPUKernel, int32_t, int64_t);
47 MS_REG_CPU_KERNEL_T_S(GatherDGrad, KernelAttr(), GatherDGradCPUKernel, int32_t, float);
48 MS_REG_CPU_KERNEL_T_S(GatherDGrad, KernelAttr(), GatherDGradCPUKernel, int32_t, float16);
49 MS_REG_CPU_KERNEL_T_S(GatherDGrad, KernelAttr(), GatherDGradCPUKernel, int32_t, bool);
50 MS_REG_CPU_KERNEL_T_S(GatherDGrad, KernelAttr(), GatherDGradCPUKernel, int64_t, int32_t);
51 MS_REG_CPU_KERNEL_T_S(GatherDGrad, KernelAttr(), GatherDGradCPUKernel, int64_t, int64_t);
52 MS_REG_CPU_KERNEL_T_S(GatherDGrad, KernelAttr(), GatherDGradCPUKernel, int64_t, float);
53 MS_REG_CPU_KERNEL_T_S(GatherDGrad, KernelAttr(), GatherDGradCPUKernel, int64_t, float16);
54 MS_REG_CPU_KERNEL_T_S(GatherDGrad, KernelAttr(), GatherDGradCPUKernel, int64_t, bool);
Dgather_d_cpu_kernel.h42 MS_REG_CPU_KERNEL_T_S(GatherD, KernelAttr(), GatherDCPUKernel, float, int32_t);
43 MS_REG_CPU_KERNEL_T_S(GatherD, KernelAttr(), GatherDCPUKernel, float, int64_t);
44 MS_REG_CPU_KERNEL_T_S(GatherD, KernelAttr(), GatherDCPUKernel, float16, int32_t);
45 MS_REG_CPU_KERNEL_T_S(GatherD, KernelAttr(), GatherDCPUKernel, float16, int64_t);
46 MS_REG_CPU_KERNEL_T_S(GatherD, KernelAttr(), GatherDCPUKernel, int32_t, int32_t);
47 MS_REG_CPU_KERNEL_T_S(GatherD, KernelAttr(), GatherDCPUKernel, int32_t, int64_t);
48 MS_REG_CPU_KERNEL_T_S(GatherD, KernelAttr(), GatherDCPUKernel, int64_t, int32_t);
49 MS_REG_CPU_KERNEL_T_S(GatherD, KernelAttr(), GatherDCPUKernel, int64_t, int64_t);
50 MS_REG_CPU_KERNEL_T_S(GatherD, KernelAttr(), GatherDCPUKernel, bool, int32_t);
51 MS_REG_CPU_KERNEL_T_S(GatherD, KernelAttr(), GatherDCPUKernel, bool, int64_t);
Dsearchsorted_cpu_kernel.h48 MS_REG_CPU_KERNEL_T_S(
53 MS_REG_CPU_KERNEL_T_S(
58 MS_REG_CPU_KERNEL_T_S(
63 MS_REG_CPU_KERNEL_T_S(
68 MS_REG_CPU_KERNEL_T_S(
73 MS_REG_CPU_KERNEL_T_S(
78 MS_REG_CPU_KERNEL_T_S(
83 MS_REG_CPU_KERNEL_T_S(
88 MS_REG_CPU_KERNEL_T_S(
93 MS_REG_CPU_KERNEL_T_S(
[all …]
Dsparse_to_dense_cpu_kernal.h44 MS_REG_CPU_KERNEL_T_S(SparseToDense,
52 MS_REG_CPU_KERNEL_T_S(SparseToDense,
60 MS_REG_CPU_KERNEL_T_S(SparseToDense,
68 MS_REG_CPU_KERNEL_T_S(SparseToDense,
76 MS_REG_CPU_KERNEL_T_S(SparseToDense,
84 MS_REG_CPU_KERNEL_T_S(SparseToDense,
92 MS_REG_CPU_KERNEL_T_S(SparseToDense,
100 MS_REG_CPU_KERNEL_T_S(SparseToDense,
108 MS_REG_CPU_KERNEL_T_S(SparseToDense,
116 MS_REG_CPU_KERNEL_T_S(SparseToDense,
[all …]
Drolling_cpu_kernel.h82 MS_REG_CPU_KERNEL_T_S(Rolling, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberT…
85 MS_REG_CPU_KERNEL_T_S(Rolling, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberT…
88 MS_REG_CPU_KERNEL_T_S(Rolling, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTyp…
91 MS_REG_CPU_KERNEL_T_S(Rolling, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTyp…
94 MS_REG_CPU_KERNEL_T_S(Rolling, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTyp…
97 MS_REG_CPU_KERNEL_T_S(Rolling, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTyp…
Dsparse_tensor_dense_matmul_cpu_kernel.h47 MS_REG_CPU_KERNEL_T_S(SparseTensorDenseMatmul,
56 MS_REG_CPU_KERNEL_T_S(SparseTensorDenseMatmul,
65 MS_REG_CPU_KERNEL_T_S(SparseTensorDenseMatmul,
74 MS_REG_CPU_KERNEL_T_S(SparseTensorDenseMatmul,
Dcpu_kernel_factory.h77 #define MS_REG_CPU_KERNEL_T_S(OPNAME, ATTR, OPCLASS, T, S) \ macro