Searched refs:cauchy_kernel (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | DistributionCauchyKernel.cu | 8 void cauchy_kernel(TensorIteratorBase& iter, double median, double sigma, std::optional<Generator> … in cauchy_kernel() function
10 at::native::templates::cuda::cauchy_kernel(iter, median, sigma, generator); in cauchy_kernel()
13 REGISTER_DISPATCH(cauchy_stub, &cauchy_kernel);
|
| D | DistributionTemplates.h | 558 void cauchy_kernel(TensorIteratorBase& iter, double median_, double sigma_, RNG gen) { in cauchy_kernel() function
574 cauchy_kernel(iter, median, sigma, check_generator<RNG>(gen)); in operator()
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | DistributionKernels.cpp | 29 static void cauchy_kernel(TensorIteratorBase& iter, double median, double sigma, std::optional<Gene… in cauchy_kernel() function
31 templates::cpu::cauchy_kernel(iter, median, sigma, generator); in cauchy_kernel()
240 REGISTER_DISPATCH(cauchy_stub, &cauchy_kernel);
|
| D | DistributionTemplates.h | 289 void cauchy_kernel(TensorIteratorBase& iter, double median, double sigma, RNG generator) { in cauchy_kernel() function
302 cauchy_kernel(iter, median, sigma, check_generator<RNG>(gen)); in operator()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | DistributionTemplates.h | 348 template<template<typename> class cauchy_kernel, typename RNG>
356 cauchy_kernel<RNG>()(iter, median, sigma, gen); in cauchy_impl_()
|
| /external/pytorch/aten/src/ATen/test/ |
| D | cpu_rng_test.cpp | 327 …native::templates::cpu::cauchy_kernel(iter, median, sigma, check_generator<TestCPUGenerator>(gen)); in TEST_F()
|