Browse Source

fix CPU StridedSliceGrad

tags/v1.6.0
zhaoting 4 years ago
parent
commit
0c9343acec
2 changed files with 2 additions and 3 deletions
  1. +2
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/slice_grad_cpu_kernel.h
  2. +0
    -3
      mindspore/ccsrc/backend/kernel_compiler/cpu/strided_slice_grad_cpu_kernel.h

+ 2
- 0
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),


+ 0
- 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



Loading…
Cancel
Save