m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

remove int32 support for TensorScatterUpdate

Browse Source
pull/13265/head
Peilin Wang 4 years ago
parent 483bb9de60
commit 121615e2ed
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File

8
mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/tensor_scatter_update_gpu_kernel.cc
Unescape View File

@ -42,14 +42,6 @@ MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate,
.AddOutputAttr(kNumberTypeInt8), .AddOutputAttr(kNumberTypeInt8),
TensorScatterUpdateGpuFwdKernel, char, int) 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, MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate,
KernelAttr() KernelAttr()
.AddInputAttr(kNumberTypeUInt8) .AddInputAttr(kNumberTypeUInt8)

Write Preview
Loading…
Cancel
Save