Home
last modified time | relevance | path

Searched refs:MS_REG_CPU_KERNEL_T (Results 1 – 25 of 53) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Darithmetic_logic_cpu_kernel.h64 MS_REG_CPU_KERNEL_T(
67 MS_REG_CPU_KERNEL_T(
70 MS_REG_CPU_KERNEL_T(
73 MS_REG_CPU_KERNEL_T(
76 MS_REG_CPU_KERNEL_T(
79 MS_REG_CPU_KERNEL_T(
82 MS_REG_CPU_KERNEL_T(
85 MS_REG_CPU_KERNEL_T(
88 MS_REG_CPU_KERNEL_T(
91 MS_REG_CPU_KERNEL_T(
[all …]
Darithmetic_cpu_kernel.h67 MS_REG_CPU_KERNEL_T(Sub, KernelAttr(), ArithmeticCPUKernel, int32_t);
68 MS_REG_CPU_KERNEL_T(Sub, KernelAttr(), ArithmeticCPUKernel, float);
69 MS_REG_CPU_KERNEL_T(Sub, KernelAttr(), ArithmeticCPUKernel, int64_t);
70 MS_REG_CPU_KERNEL_T(Pow, KernelAttr(), ArithmeticCPUKernel, int32_t);
71 MS_REG_CPU_KERNEL_T(Pow, KernelAttr(), ArithmeticCPUKernel, float);
72 MS_REG_CPU_KERNEL_T(Pow, KernelAttr(), ArithmeticCPUKernel, int64_t);
73 MS_REG_CPU_KERNEL_T(RealDiv, KernelAttr(), ArithmeticCPUKernel, int32_t);
74 MS_REG_CPU_KERNEL_T(RealDiv, KernelAttr(), ArithmeticCPUKernel, float);
75 MS_REG_CPU_KERNEL_T(RealDiv, KernelAttr(), ArithmeticCPUKernel, float16);
76 MS_REG_CPU_KERNEL_T(RealDiv, KernelAttr(), ArithmeticCPUKernel, int64_t);
[all …]
Dreduce_cpu_kernel.h49 MS_REG_CPU_KERNEL_T(ReduceMean, KernelAttr(), ReduceCPUKernel, float);
50 MS_REG_CPU_KERNEL_T(ReduceMean, KernelAttr(), ReduceCPUKernel, double);
51 MS_REG_CPU_KERNEL_T(ReduceMean, KernelAttr(), ReduceCPUKernel, int32_t);
52 MS_REG_CPU_KERNEL_T(ReduceMean, KernelAttr(), ReduceCPUKernel, int64_t);
54 MS_REG_CPU_KERNEL_T(ReduceMax, KernelAttr(), ReduceCPUKernel, float);
55 MS_REG_CPU_KERNEL_T(ReduceMax, KernelAttr(), ReduceCPUKernel, double);
56 MS_REG_CPU_KERNEL_T(ReduceMax, KernelAttr(), ReduceCPUKernel, int32_t);
57 MS_REG_CPU_KERNEL_T(ReduceMax, KernelAttr(), ReduceCPUKernel, int64_t);
59 MS_REG_CPU_KERNEL_T(ReduceSum, KernelAttr(), ReduceCPUKernel, float);
60 MS_REG_CPU_KERNEL_T(ReduceSum, KernelAttr(), ReduceCPUKernel, double);
[all …]
Dgather_cpu_kernel.h48 MS_REG_CPU_KERNEL_T(
51 MS_REG_CPU_KERNEL_T(
54 MS_REG_CPU_KERNEL_T(
57 MS_REG_CPU_KERNEL_T(
60 MS_REG_CPU_KERNEL_T(
63 MS_REG_CPU_KERNEL_T(
66 MS_REG_CPU_KERNEL_T(
69 MS_REG_CPU_KERNEL_T(
72 MS_REG_CPU_KERNEL_T(
76 MS_REG_CPU_KERNEL_T(
[all …]
Dscatter_arithmetic_cpu_kernel.h61 MS_REG_CPU_KERNEL_T(ScatterAdd,
68 MS_REG_CPU_KERNEL_T(ScatterAdd,
75 MS_REG_CPU_KERNEL_T(ScatterAdd,
82 MS_REG_CPU_KERNEL_T(ScatterSub,
89 MS_REG_CPU_KERNEL_T(ScatterSub,
96 MS_REG_CPU_KERNEL_T(ScatterSub,
103 MS_REG_CPU_KERNEL_T(ScatterMul,
110 MS_REG_CPU_KERNEL_T(ScatterMul,
117 MS_REG_CPU_KERNEL_T(ScatterMul,
124 MS_REG_CPU_KERNEL_T(ScatterDiv,
[all …]
Dpack_cpu_kernel.h47 MS_REG_CPU_KERNEL_T(Stack, KernelAttr(), PackCpuFwdKernel, int8_t)
48 MS_REG_CPU_KERNEL_T(Stack, KernelAttr(), PackCpuFwdKernel, int16_t)
49 MS_REG_CPU_KERNEL_T(Stack, KernelAttr(), PackCpuFwdKernel, int32_t)
50 MS_REG_CPU_KERNEL_T(Stack, KernelAttr(), PackCpuFwdKernel, int64_t)
51 MS_REG_CPU_KERNEL_T(Stack, KernelAttr(), PackCpuFwdKernel, uint8_t)
52 MS_REG_CPU_KERNEL_T(Stack, KernelAttr(), PackCpuFwdKernel, uint16_t)
53 MS_REG_CPU_KERNEL_T(Stack, KernelAttr(), PackCpuFwdKernel, uint32_t)
54 MS_REG_CPU_KERNEL_T(Stack, KernelAttr(), PackCpuFwdKernel, uint64_t)
55 MS_REG_CPU_KERNEL_T(Stack, KernelAttr(), PackCpuFwdKernel, float16)
56 MS_REG_CPU_KERNEL_T(Stack, KernelAttr(), PackCpuFwdKernel, float)
[all …]
Dgathernd_cpu_kernel.h50 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, bool);
51 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, int8_t);
52 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, int16_t);
53 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, int32_t);
54 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, int64_t);
55 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, uint8_t);
56 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, uint16_t);
57 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, uint32_t);
58 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, uint64_t);
59 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, float);
[all …]
Dconcat_cpu_kernel.h44 MS_REG_CPU_KERNEL_T(Concat, KernelAttr(), ConcatCPUKernel, float);
45 MS_REG_CPU_KERNEL_T(Concat, KernelAttr(), ConcatCPUKernel, int8_t)
46 MS_REG_CPU_KERNEL_T(Concat, KernelAttr(), ConcatCPUKernel, int16_t)
47 MS_REG_CPU_KERNEL_T(Concat, KernelAttr(), ConcatCPUKernel, int32_t)
48 MS_REG_CPU_KERNEL_T(Concat, KernelAttr(), ConcatCPUKernel, int64_t)
49 MS_REG_CPU_KERNEL_T(Concat, KernelAttr(), ConcatCPUKernel, uint8_t)
50 MS_REG_CPU_KERNEL_T(Concat, KernelAttr(), ConcatCPUKernel, uint16_t)
51 MS_REG_CPU_KERNEL_T(Concat, KernelAttr(), ConcatCPUKernel, uint32_t)
52 MS_REG_CPU_KERNEL_T(Concat, KernelAttr(), ConcatCPUKernel, uint64_t)
53 MS_REG_CPU_KERNEL_T(Concat, KernelAttr(), ConcatCPUKernel, bool)
Dcrop_and_resize_cpu_kernel.h65 MS_REG_CPU_KERNEL_T(CropAndResize,
74 MS_REG_CPU_KERNEL_T(CropAndResize,
83 MS_REG_CPU_KERNEL_T(CropAndResize,
92 MS_REG_CPU_KERNEL_T(CropAndResize,
101 MS_REG_CPU_KERNEL_T(CropAndResize,
110 MS_REG_CPU_KERNEL_T(CropAndResize,
119 MS_REG_CPU_KERNEL_T(CropAndResize,
128 MS_REG_CPU_KERNEL_T(CropAndResize,
137 MS_REG_CPU_KERNEL_T(CropAndResize,
146 MS_REG_CPU_KERNEL_T(CropAndResize,
[all …]
Dunpack_cpu_kernel.h49 MS_REG_CPU_KERNEL_T(Unstack,
52 MS_REG_CPU_KERNEL_T(Unstack,
55 MS_REG_CPU_KERNEL_T(Unstack,
58 MS_REG_CPU_KERNEL_T(Unstack,
61 MS_REG_CPU_KERNEL_T(Unstack,
64 MS_REG_CPU_KERNEL_T(Unstack,
67 MS_REG_CPU_KERNEL_T(Unstack,
70 MS_REG_CPU_KERNEL_T(Unstack,
73 MS_REG_CPU_KERNEL_T(Unstack,
76 MS_REG_CPU_KERNEL_T(
[all …]
Dprint_cpu_kernel.h47 MS_REG_CPU_KERNEL_T(Print,
50 MS_REG_CPU_KERNEL_T(Print,
53 MS_REG_CPU_KERNEL_T(Print,
56 MS_REG_CPU_KERNEL_T(Print,
59 MS_REG_CPU_KERNEL_T(Print,
62 MS_REG_CPU_KERNEL_T(Print,
65 MS_REG_CPU_KERNEL_T(Print,
68 MS_REG_CPU_KERNEL_T(Print,
71 MS_REG_CPU_KERNEL_T(Print,
74 MS_REG_CPU_KERNEL_T(Print,
[all …]
Dspacetodepth_cpu_kernel.h42 MS_REG_CPU_KERNEL_T(
46 MS_REG_CPU_KERNEL_T(
50 MS_REG_CPU_KERNEL_T(SpaceToDepth,
54 MS_REG_CPU_KERNEL_T(SpaceToDepth,
58 MS_REG_CPU_KERNEL_T(SpaceToDepth,
62 MS_REG_CPU_KERNEL_T(SpaceToDepth,
66 MS_REG_CPU_KERNEL_T(SpaceToDepth,
70 MS_REG_CPU_KERNEL_T(SpaceToDepth,
74 MS_REG_CPU_KERNEL_T(SpaceToDepth,
78 MS_REG_CPU_KERNEL_T(SpaceToDepth,
Ddepthtospace_cpu_kernel.h43 MS_REG_CPU_KERNEL_T(
47 MS_REG_CPU_KERNEL_T(
51 MS_REG_CPU_KERNEL_T(DepthToSpace,
55 MS_REG_CPU_KERNEL_T(DepthToSpace,
59 MS_REG_CPU_KERNEL_T(DepthToSpace,
63 MS_REG_CPU_KERNEL_T(DepthToSpace,
67 MS_REG_CPU_KERNEL_T(DepthToSpace,
71 MS_REG_CPU_KERNEL_T(DepthToSpace,
75 MS_REG_CPU_KERNEL_T(DepthToSpace,
79 MS_REG_CPU_KERNEL_T(DepthToSpace,
Deltwise_grad_cpu_kernel.h58 MS_REG_CPU_KERNEL_T(
62 MS_REG_CPU_KERNEL_T(
66 MS_REG_CPU_KERNEL_T(
70 MS_REG_CPU_KERNEL_T(
74 MS_REG_CPU_KERNEL_T(
78 MS_REG_CPU_KERNEL_T(
82 MS_REG_CPU_KERNEL_T(GeLUGrad,
89 MS_REG_CPU_KERNEL_T(
93 MS_REG_CPU_KERNEL_T(
97 MS_REG_CPU_KERNEL_T(
[all …]
Darithmetic_self_cpu_kernel.h115 MS_REG_CPU_KERNEL_T(Identity, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTyp…
117 MS_REG_CPU_KERNEL_T(Identity, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberType…
119 MS_REG_CPU_KERNEL_T(Identity, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumberTyp…
121 MS_REG_CPU_KERNEL_T(Identity, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberType…
123 MS_REG_CPU_KERNEL_T(Identity, KernelAttr().AddInputAttr(kNumberTypeUInt16).AddOutputAttr(kNumberTyp…
125 MS_REG_CPU_KERNEL_T(Identity, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumberType…
127 MS_REG_CPU_KERNEL_T(Identity, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberType…
129 MS_REG_CPU_KERNEL_T(Identity, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeI…
131 MS_REG_CPU_KERNEL_T(Identity, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberTy…
133 MS_REG_CPU_KERNEL_T(Identity, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTy…
[all …]
Dsplit_cpu_kernel.h55 MS_REG_CPU_KERNEL_T(Split, KernelAttr(), SplitCPUKernel, float);
56 MS_REG_CPU_KERNEL_T(Split, KernelAttr(), SplitCPUKernel, float16);
57 MS_REG_CPU_KERNEL_T(Split, KernelAttr(), SplitCPUKernel, double);
58 MS_REG_CPU_KERNEL_T(Split, KernelAttr(), SplitCPUKernel, int32_t);
59 MS_REG_CPU_KERNEL_T(Split, KernelAttr(), SplitCPUKernel, uint32_t);
60 MS_REG_CPU_KERNEL_T(Split, KernelAttr(), SplitCPUKernel, int64_t);
Dmasked_select_cpu_kernel.h46 MS_REG_CPU_KERNEL_T(
51 MS_REG_CPU_KERNEL_T(
56 MS_REG_CPU_KERNEL_T(
61 MS_REG_CPU_KERNEL_T(
66 MS_REG_CPU_KERNEL_T(
71 MS_REG_CPU_KERNEL_T(
Dhsigmoid_grad_cpu_kernel.h44 MS_REG_CPU_KERNEL_T(HSigmoidGrad, KernelAttr(), HSigmoidGradCPUKernel, int8_t);
45 MS_REG_CPU_KERNEL_T(HSigmoidGrad, KernelAttr(), HSigmoidGradCPUKernel, int16_t);
46 MS_REG_CPU_KERNEL_T(HSigmoidGrad, KernelAttr(), HSigmoidGradCPUKernel, int32_t);
47 MS_REG_CPU_KERNEL_T(HSigmoidGrad, KernelAttr(), HSigmoidGradCPUKernel, int64_t);
48 MS_REG_CPU_KERNEL_T(HSigmoidGrad, KernelAttr(), HSigmoidGradCPUKernel, float);
Dhswish_cpu_kernel.h44 MS_REG_CPU_KERNEL_T(HSwish, KernelAttr(), HSwishCPUKernel, int8_t);
45 MS_REG_CPU_KERNEL_T(HSwish, KernelAttr(), HSwishCPUKernel, int16_t);
46 MS_REG_CPU_KERNEL_T(HSwish, KernelAttr(), HSwishCPUKernel, int32_t);
47 MS_REG_CPU_KERNEL_T(HSwish, KernelAttr(), HSwishCPUKernel, int64_t);
48 MS_REG_CPU_KERNEL_T(HSwish, KernelAttr(), HSwishCPUKernel, float);
Dhswish_grad_cpu_kernel.h44 MS_REG_CPU_KERNEL_T(HSwishGrad, KernelAttr(), HSwishGradCPUKernel, int8_t);
45 MS_REG_CPU_KERNEL_T(HSwishGrad, KernelAttr(), HSwishGradCPUKernel, int16_t);
46 MS_REG_CPU_KERNEL_T(HSwishGrad, KernelAttr(), HSwishGradCPUKernel, int32_t);
47 MS_REG_CPU_KERNEL_T(HSwishGrad, KernelAttr(), HSwishGradCPUKernel, int64_t);
48 MS_REG_CPU_KERNEL_T(HSwishGrad, KernelAttr(), HSwishGradCPUKernel, float);
Dhsigmoid_cpu_kernel.h44 MS_REG_CPU_KERNEL_T(HSigmoid, KernelAttr(), HSigmoidCPUKernel, int8_t);
46 MS_REG_CPU_KERNEL_T(HSigmoid, KernelAttr(), HSigmoidCPUKernel, int16_t);
48 MS_REG_CPU_KERNEL_T(HSigmoid, KernelAttr(), HSigmoidCPUKernel, int32_t);
50 MS_REG_CPU_KERNEL_T(HSigmoid, KernelAttr(), HSigmoidCPUKernel, int64_t);
52 MS_REG_CPU_KERNEL_T(HSigmoid, KernelAttr(), HSigmoidCPUKernel, float);
Dmasked_select_grad_cpu_kernel.h49 MS_REG_CPU_KERNEL_T(MaskedSelectGrad,
57 MS_REG_CPU_KERNEL_T(MaskedSelectGrad,
65 MS_REG_CPU_KERNEL_T(MaskedSelectGrad,
73 MS_REG_CPU_KERNEL_T(MaskedSelectGrad,
81 MS_REG_CPU_KERNEL_T(MaskedSelectGrad,
89 MS_REG_CPU_KERNEL_T(MaskedSelectGrad,
Dshift_cpu_kernel.h54 MS_REG_CPU_KERNEL_T(
58 MS_REG_CPU_KERNEL_T(
63 MS_REG_CPU_KERNEL_T(
68 MS_REG_CPU_KERNEL_T(
72 MS_REG_CPU_KERNEL_T(
Dmaximum_cpu_kernel.h77 MS_REG_CPU_KERNEL_T(Maximum, KernelAttr(), MaximumCPUKernel, int32_t);
78 MS_REG_CPU_KERNEL_T(Maximum, KernelAttr(), MaximumCPUKernel, int64_t);
79 MS_REG_CPU_KERNEL_T(Maximum, KernelAttr(), MaximumCPUKernel, uint32_t);
80 MS_REG_CPU_KERNEL_T(Maximum, KernelAttr(), MaximumCPUKernel, uint64_t);
81 MS_REG_CPU_KERNEL_T(Maximum, KernelAttr(), MaximumCPUKernel, float);
82 MS_REG_CPU_KERNEL_T(Maximum, KernelAttr(), MaximumCPUKernel, double);
Dminimum_cpu_kernel.h77 MS_REG_CPU_KERNEL_T(Minimum, KernelAttr(), MinimumCPUKernel, int32_t);
78 MS_REG_CPU_KERNEL_T(Minimum, KernelAttr(), MinimumCPUKernel, uint32_t);
79 MS_REG_CPU_KERNEL_T(Minimum, KernelAttr(), MinimumCPUKernel, float);
80 MS_REG_CPU_KERNEL_T(Minimum, KernelAttr(), MinimumCPUKernel, int64_t);
81 MS_REG_CPU_KERNEL_T(Minimum, KernelAttr(), MinimumCPUKernel, uint64_t);
82 MS_REG_CPU_KERNEL_T(Minimum, KernelAttr(), MinimumCPUKernel, double);

123