Home
last modified time | relevance | path

Searched defs:AtomicFPOp (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DAtomic.cuh17 struct AtomicFPOp<at::Half> { struct
39 struct AtomicFPOp<at::BFloat16> { argument
61 struct AtomicFPOp<double> { struct
63 inline __device__ double operator() (double * address, double val, const func_t& func) { in operator ()()