Searched defs:MsAtomicAdd (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | util.cuh | 207 __device__ __forceinline__ T MsAtomicAdd(T *address, T val) { in MsAtomicAdd() function
213 __device__ __forceinline__ int MsAtomicAdd(int *address, int val) { in MsAtomicAdd() function
218 __device__ __forceinline__ unsigned int MsAtomicAdd(unsigned int *address, unsigned int val) { in MsAtomicAdd() function
223 __device__ __forceinline__ unsigned long long int MsAtomicAdd(unsigned long long int *address, // … in MsAtomicAdd() function
229 __device__ __forceinline__ float MsAtomicAdd(float *address, float val) { in MsAtomicAdd() function
234 __device__ __forceinline__ bool MsAtomicAdd(bool *address, bool val) { in MsAtomicAdd() function
240 __device__ __forceinline__ Complex<float> MsAtomicAdd(Complex<float> *address, Complex<float> val) { in MsAtomicAdd() function
246 __device__ __forceinline__ Complex<double> MsAtomicAdd(Complex<double> *address, Complex<double> va… in MsAtomicAdd() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | deformable_offsets_grad.cc | 122 void MsAtomicAdd(T *output_grad_x, const size_t &output_grad_base_pos, const T &added_value) { in MsAtomicAdd() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | deformable_offsets_grad_cpu_kernel.cc | 118 void MsAtomicAdd(T *output_grad_x, const size_t &output_grad_base_pos, const T &added_value) { in MsAtomicAdd() function
|