Searched defs:rng_engine_inputs (Results 1 – 10 of 10) sorted by relevance
151 PhiloxCudaState rng_engine_inputs; in launch_poisson_cuda_kernel() local164 PhiloxCudaState rng_engine_inputs; in launch_binomial_cuda_kernel() local177 PhiloxCudaState rng_engine_inputs; in launch_gamma_kernel() local
227 const PhiloxCudaState rng_engine_inputs, in launcher()352 PhiloxCudaState rng_engine_inputs; in dropout_cuda() local
130 PhiloxCudaState rng_engine_inputs; in distribution_nullary_kernel() local627 PhiloxCudaState rng_engine_inputs; in bernoulli_kernel() local
46 at::PhiloxCudaState rng_engine_inputs; in randperm_handle_duplicate_keys() local
92 PhiloxCudaState rng_engine_inputs; in _rrelu_with_noise_cuda_train() local
419 PhiloxCudaState rng_engine_inputs; in multinomial_with_replacement_kernel_impl() local
1338 at::PhiloxCudaState rng_engine_inputs, in rand_uniform_kernel()1400 at::PhiloxCudaState rng_engine_inputs; in _fill_mem_eff_dropout_mask_() local
395 at::PhiloxCudaState rng_engine_inputs; in _efficient_attention_backward() local
192 at::PhiloxCudaState rng_engine_inputs = at::PhiloxCudaState(0, 0); member
677 at::PhiloxCudaState rng_engine_inputs = {0, 0}; member