Searched refs:CUDAHistogramMemoryType (Results 1 – 1 of 1) sorted by relevance
32 enum class CUDAHistogramMemoryType { SHARED, GLOBAL }; enum62 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()