Browse Source

remove int32 support for TensorScatterUpdate

tags/v1.2.0-rc1
Peilin Wang 4 years ago
parent
commit
121615e2ed
1 changed files with 0 additions and 8 deletions
  1. +0
    -8
      mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/tensor_scatter_update_gpu_kernel.cc

+ 0
- 8
mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/tensor_scatter_update_gpu_kernel.cc View File

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


Loading…
Cancel
Save