Searched defs:cuda_gen (Results 1 – 3 of 3) sorted by relevance
75 auto cuda_gen = check_generator<CUDAGeneratorImpl>(gen); in createCUDAGenerator() local380 c10::intrusive_ptr<CUDAGeneratorImpl> cuda_gen = in graphsafe_set_state() local
77 c10::intrusive_ptr<CUDAGeneratorImpl> cuda_gen = in register_generator_state() local
527 auto cuda_gen = globalContext().defaultGenerator( in manual_seed() local