|
|
|
@@ -42,14 +42,6 @@ MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, |
|
|
|
.AddOutputAttr(kNumberTypeInt8), |
|
|
|
TensorScatterUpdateGpuFwdKernel, char, int) |
|
|
|
|
|
|
|
MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, |
|
|
|
KernelAttr() |
|
|
|
.AddInputAttr(kNumberTypeInt32) |
|
|
|
.AddInputAttr(kNumberTypeInt32) |
|
|
|
.AddInputAttr(kNumberTypeInt32) |
|
|
|
.AddOutputAttr(kNumberTypeInt32), |
|
|
|
TensorScatterUpdateGpuFwdKernel, int, int) |
|
|
|
|
|
|
|
MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, |
|
|
|
KernelAttr() |
|
|
|
.AddInputAttr(kNumberTypeUInt8) |
|
|
|
|