Searched refs:launch_dirichlet_kernel (Results 1 – 3 of 3) sorted by relevance
19 void launch_dirichlet_kernel(TensorIteratorBase &iter);
57 launch_dirichlet_kernel(iter); in _s_dirichlet_cuda()
133 void launch_dirichlet_kernel(at::TensorIteratorBase &iter) { in launch_dirichlet_kernel() function