Home
last modified time | relevance | path

Searched refs:CUDAHistogramMemoryType (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSummaryOps.cu32 enum class CUDAHistogramMemoryType { SHARED, GLOBAL }; enum
62 CUDAHistogramMemoryType MemoryType,
77 if (MemoryType == CUDAHistogramMemoryType::SHARED) { in C10_LAUNCH_BOUNDS_1()
150 case CUDAHistogramMemoryType::SHARED: \
151 HANDLE_CASE(CUDAHistogramMemoryType::SHARED, getOp, sharedMem); \
154 HANDLE_CASE(CUDAHistogramMemoryType::GLOBAL, getOp, 0); \
200 CUDAHistogramMemoryType memType = CUDAHistogramMemoryType::GLOBAL; in CUDA_tensor_histogram()
217 memType = CUDAHistogramMemoryType::SHARED; in CUDA_tensor_histogram()