|
|
|
@@ -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), |
|
|
|
|