Searched refs:bernoulli_tensor_cuda_kernel (Results 1 – 1 of 1) sorted by relevance
581 void bernoulli_tensor_cuda_kernel( in bernoulli_tensor_cuda_kernel() function641 return bernoulli_tensor_cuda_kernel<double, double>(self, *p, rng_engine_inputs); in bernoulli_kernel()643 return bernoulli_tensor_cuda_kernel<scalar_t, float>(self, *p, rng_engine_inputs); in bernoulli_kernel()