p27049618
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix CPU StridedSliceGrad

This commit is contained in:
zhaoting 2021-09-30 09:36:37 +08:00
parent 9cf07ea45d
commit 0c9343acec
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/kernel_compiler/cpu/slice_grad_cpu_kernel.h
View File

@ -73,6 +73,8 @@ MS_REG_CPU_KERNEL(
SliceGrad, KernelAttr().AddInputAttr(kNumberTypeBool).AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumberTypeBool),
SliceGradCPUKernel);
MS_REG_CPU_KERNEL(StridedSliceGrad, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32),
SliceGradCPUKernel);
MS_REG_CPU_KERNEL(StridedSliceGrad, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt32),
SliceGradCPUKernel);
MS_REG_CPU_KERNEL(StridedSliceGrad, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberTypeFloat64),

3
mindspore/ccsrc/backend/kernel_compiler/cpu/strided_slice_grad_cpu_kernel.h
View File

@ -46,9 +46,6 @@ class StridedSliceGradCPUKernel : public CPUKernel {
StridedSliceParameter *param_{nullptr};
};
MS_REG_CPU_KERNEL(StridedSliceGrad, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32),
StridedSliceGradCPUKernel);
} // namespace kernel
} // namespace mindspore