Searched refs:num_gpu_init_flag (Results 1 – 1 of 1) sorted by relevance
18 static c10::once_flag num_gpu_init_flag; variable50 c10::call_once(num_gpu_init_flag, initCUDAGenVector); in getDefaultCUDAGenerator()68 c10::call_once(num_gpu_init_flag, initCUDAGenVector); in createCUDAGenerator()