Home
last modified time | relevance | path

Searched refs:TensorScatterSub (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dtensor_scatter_sub_gpu_kernel.cc22 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,
30 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,
38 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,
46 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,
54 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,
63 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,
71 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,
79 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,
87 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,
95 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,
Dtensor_scatter_sub_gpu_kernel.h86 …TensorScatterSub(input, indices, update, output, block_size_, update_size, output_size, indices_di… in Launch()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dtensor_scatter_sub.cu50 void TensorScatterSub(T *input, S *indices, T *update, T *output, const size_t &block_size, const s… in TensorScatterSub() function
60 template void TensorScatterSub<half, int>(half *input, int *indices, half *update, half *output,
65 template void TensorScatterSub<float, int>(float *input, int *indices, float *update, float *output,
71 template void TensorScatterSub<char, int>(char *input, int *indices, char *update, char *output,
76 template void TensorScatterSub<unsigned char, int>(unsigned char *input, int *indices, unsigned cha…
82 template void TensorScatterSub<int, int>(int *input, int *indices, int *update, int *output, const …
88 template void TensorScatterSub<half, int64_t>(half *input, int64_t *indices, half *update, half *ou…
94 template void TensorScatterSub<float, int64_t>(float *input, int64_t *indices, float *update, float…
100 template void TensorScatterSub<char, int64_t>(char *input, int64_t *indices, char *update, char *ou…
106 template void TensorScatterSub<unsigned char, int64_t>(unsigned char *input, int64_t *indices, unsi…
[all …]
Dtensor_scatter_sub.cuh23 void TensorScatterSub(T *input, S *indices, T *update, T *output, const size_t &block_size, const s…
/third_party/mindspore/mindspore/ops/_grad_experimental/
Dgrad_array_ops.py49 @bprop_getters.register(P.TensorScatterSub)
/third_party/mindspore/tests/st/ops/gpu/
Dtest_tensor_scatter_sub.py26 self.scatter = P.TensorScatterSub()
/third_party/mindspore/mindspore/ops/operations/
D__init__.py37 TensorScatterMax, TensorScatterMin, TensorScatterSub)
Darray_ops.py6349 class TensorScatterSub(PrimitiveWithInfer): class