Searched defs:cuda_guard (Results 1 – 1 of 1) sorted by relevance
286 at::cuda::OptionalCUDAStreamGuard cuda_guard; in _broadcast_out_impl() local335 at::cuda::OptionalCUDAStreamGuard cuda_guard; in _broadcast_out_impl() local