Searched defs:philox_args (Results 1 – 8 of 8) sorted by relevance
47 at::PhiloxCudaState philox_args) { in poisson_cuda_kernel()79 at::PhiloxCudaState philox_args) { in binomial_cuda_kernel()100 at::PhiloxCudaState philox_args) { in gamma_cuda_kernel()
13 …le_duplicate_keys_kernel(T *keys, scalar_t *data, T mask, int n, at::PhiloxCudaState philox_args) { in randperm_handle_duplicate_keys_kernel()
182 PhiloxCudaState philox_args, in distribution_binary_elementwise_kernel()235 void distribution_binary_kernel(TensorIteratorBase &iter, PhiloxCudaState philox_args, const func_t… in distribution_binary_kernel()583 PhiloxCudaState philox_args) { in bernoulli_tensor_cuda_kernel()
52 PhiloxCudaState philox_args) { in fused_dropout_kernel_vec()140 PhiloxCudaState philox_args) { in fused_dropout_kernel()
25 PhiloxCudaState philox_args, in rrelu_with_noise_cuda_kernel()
141 sampleMultinomialWithReplacement(PhiloxCudaState philox_args, in sampleMultinomialWithReplacement()
122 at::PhiloxCudaState philox_args; member
985 at::PhiloxCudaState philox_args; in mha_bwd() local1219 at::PhiloxCudaState philox_args; in mha_varlen_bwd() local