Searched defs:CUDAStreamCaptureModeGuard (Results 1 – 1 of 1) sorted by relevance
20 struct C10_CUDA_API CUDAStreamCaptureModeGuard { struct21 CUDAStreamCaptureModeGuard(cudaStreamCaptureMode desired) in CUDAStreamCaptureModeGuard() argument25 ~CUDAStreamCaptureModeGuard() { in ~CUDAStreamCaptureModeGuard() argument