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