Searched defs:MsAtomicBinaryOpImpl (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | util.cuh | 38 struct MsAtomicBinaryOpImpl<Func, T, OneByte> { struct
39 __device__ __forceinline__ T operator()(T *address, T val) { in operator ()()
89 struct MsAtomicBinaryOpImpl<Func, T, TwoByte> { struct
90 __device__ __forceinline__ T operator()(T *address, T val) { in operator ()()
123 struct MsAtomicBinaryOpImpl<Func, T, FourByte> { struct
124 __device__ __forceinline__ T operator()(T *address, T val) { in operator ()()
143 struct MsAtomicBinaryOpImpl<Func, T, EightByte> { struct
144 __device__ __forceinline__ T operator()(T *address, T val) { in operator ()()
|