Searched refs:self_data_start (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ScatterGatherKernel.cu | 24 …constexpr C10_DEVICE void operator() (scalar_t* self_data_start, int64_t index, int64_t numel, con… in operator ()() argument
26 gpuAtomicMul(self_data_start + index, *src_data); in operator ()()
34 …constexpr C10_DEVICE void operator() (scalar_t* self_data_start, int64_t index, int64_t numel, con… in operator ()() argument
35 fastAtomicAdd(self_data_start, index, numel, *src_data, true); in operator ()()
43 …constexpr C10_DEVICE void operator() (scalar_t* self_data_start, int64_t index, int64_t numel, con… in operator ()() argument
44 fastAtomicAdd(self_data_start, index, numel, *src_data, true); in operator ()()
52 …constexpr C10_DEVICE void operator() (scalar_t* self_data_start, int64_t index, int64_t numel, con… in operator ()() argument
54 gpuAtomicMin(self_data_start + index, *src_data); in operator ()()
62 …constexpr C10_DEVICE void operator() (scalar_t* self_data_start, int64_t index, int64_t numel, con… in operator ()() argument
64 gpuAtomicMax(self_data_start + index, *src_data); in operator ()()
[all …]
|
| H A D | Indexing.cu | 297 …constexpr C10_DEVICE void operator() (scalar_t* self_data_start, int64_t index, int64_t numel, con… in operator ()() argument
299 gpuAtomicMul(self_data_start + index, *src_data); in operator ()()
307 …constexpr C10_DEVICE void operator() (scalar_t* self_data_start, int64_t index, int64_t numel, con… in operator ()() argument
308 fastAtomicAdd(self_data_start, index, numel, *src_data, true); in operator ()()
316 …constexpr C10_DEVICE void operator() (scalar_t* self_data_start, int64_t index, int64_t numel, con… in operator ()() argument
318 gpuAtomicMin(self_data_start + index, *src_data); in operator ()()
326 …constexpr C10_DEVICE void operator() (scalar_t* self_data_start, int64_t index, int64_t numel, con… in operator ()() argument
328 gpuAtomicMax(self_data_start + index, *src_data); in operator ()()
|