Searched refs:TensorScatterMin (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | tensor_scatter_min_gpu_kernel.cc | 22 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,
30 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,
38 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,
46 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,
54 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,
63 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,
71 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,
79 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,
87 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,
95 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,
|
| D | tensor_scatter_min_gpu_kernel.h | 85 …TensorScatterMin(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_min.cu | 50 void TensorScatterMin(T *input, S *indices, T *update, T *output, const size_t &block_size, const s… in TensorScatterMin() function
60 template void TensorScatterMin<half, int>(half *input, int *indices, half *update, half *output,
66 template void TensorScatterMin<float, int>(float *input, int *indices, float *update, float *output,
72 template void TensorScatterMin<char, int>(char *input, int *indices, char *update, char *output,
78 template void TensorScatterMin<unsigned char, int>(unsigned char *input, int *indices, unsigned cha…
84 template void TensorScatterMin<int, int>(int *input, int *indices, int *update, int *output,
91 template void TensorScatterMin<half, int64_t>(half *input, int64_t *indices, half *update, half *ou…
97 template void TensorScatterMin<float, int64_t>(float *input, int64_t *indices, float *update, float…
103 template void TensorScatterMin<char, int64_t>(char *input, int64_t *indices, char *update, char *ou…
109 template void TensorScatterMin<unsigned char, int64_t>(unsigned char *input, int64_t *indices, unsi…
[all …]
|
| D | tensor_scatter_min.cuh | 23 void TensorScatterMin(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_min.py | 26 self.scatter = P.TensorScatterMin()
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_array_ops.py | 89 @bprop_getters.register(P.TensorScatterMin)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 37 TensorScatterMax, TensorScatterMin, TensorScatterSub)
|
| D | array_ops.py | 6265 class TensorScatterMin(PrimitiveWithInfer): class
|