Searched defs:GPU_LAMBDA (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ | ||
H A D | thread_constants.h | 8 #define GPU_LAMBDA __host__ __device__ macro |
H A D | RangeFactories.cu | 22 #define GPU_LAMBDA __device__ __host__ macro |