Searched refs:TensorScatterMax (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | tensor_scatter_max_gpu_kernel.cc | 22 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,
30 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,
38 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,
46 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,
54 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,
63 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,
71 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,
79 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,
87 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,
95 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,
|
| D | tensor_scatter_max_gpu_kernel.h | 86 …TensorScatterMax(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_max.cu | 50 void TensorScatterMax(T *input, S *indices, T *update, T *output, const size_t &block_size, const s… in TensorScatterMax() function
60 template void TensorScatterMax<half, int>(half *input, int *indices, half *update, half *output,
65 template void TensorScatterMax<float, int>(float *input, int *indices, float *update, float *output,
71 template void TensorScatterMax<char, int>(char *input, int *indices, char *update, char *output,
76 template void TensorScatterMax<unsigned char, int>(unsigned char *input, int *indices, unsigned cha…
82 template void TensorScatterMax<int, int>(int *input, int *indices, int *update, int *output, const …
88 template void TensorScatterMax<half, int64_t>(half *input, int64_t *indices, half *update, half *ou…
94 template void TensorScatterMax<float, int64_t>(float *input, int64_t *indices, float *update, float…
100 template void TensorScatterMax<char, int64_t>(char *input, int64_t *indices, char *update, char *ou…
106 template void TensorScatterMax<unsigned char, int64_t>(unsigned char *input, int64_t *indices, unsi…
[all …]
|
| D | tensor_scatter_max.cuh | 23 void TensorScatterMax(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_max.py | 26 self.scatter = P.TensorScatterMax()
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_array_ops.py | 80 @bprop_getters.register(P.TensorScatterMax)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 37 TensorScatterMax, TensorScatterMin, TensorScatterSub)
|
| D | array_ops.py | 6181 class TensorScatterMax(PrimitiveWithInfer): class
|