Searched refs:TensorScatterAdd (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | tensor_scatter_add_gpu_kernel.cc | 21 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd,
28 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd,
35 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd,
42 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd,
49 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd,
56 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd,
63 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd,
|
| D | tensor_scatter_add_gpu_kernel.h | 89 …TensorScatterAdd(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/ |
| D | tensor_scatter_add.cu | 50 void TensorScatterAdd(T *input, S *indices, T *update, T *output, const size_t &block_size, const s… in TensorScatterAdd() function
59 template void TensorScatterAdd<half, int>(half *input, int *indices, half *update, half *output,
63 template void TensorScatterAdd<float, int>(float *input, int *indices, float *update, float *output,
68 template void TensorScatterAdd<double, int>(double *input, int *indices, double *update, double *ou…
73 template void TensorScatterAdd<char, int>(char *input, int *indices, char *update, char *output,
77 template void TensorScatterAdd<unsigned char, int>(unsigned char *input, int *indices, unsigned cha…
82 template void TensorScatterAdd<int, int>(int *input, int *indices, int *update, int *output, const …
86 template void TensorScatterAdd<double, int64_t>(double *input, int64_t *indices, double *update, do…
|
| D | tensor_scatter_add.cuh | 23 void TensorScatterAdd(T *input, S *indices, T *update, T *output, const size_t &block_size, const s…
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_tensor_scatter_add.py | 26 self.scatter_add = P.TensorScatterAdd()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 32 … Squeeze, StridedSlice, Tile, TensorScatterUpdate, TensorScatterAdd, EditDistance, Sort,
|
| D | array_ops.py | 3852 class TensorScatterAdd(PrimitiveWithInfer): class
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 220 tensor_scatter_add = P.TensorScatterAdd()
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 782 @bprop_getters.register(P.TensorScatterAdd)
|