Searched refs:log_normal_kernel (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DistributionLogNormalKernel.cu | 8 void log_normal_kernel(TensorIteratorBase& iter, double mean, double std, std::optional<Generator> … in log_normal_kernel() function
10 at::native::templates::cuda::log_normal_kernel(iter, mean, std, generator); in log_normal_kernel()
13 REGISTER_DISPATCH(log_normal_stub, &log_normal_kernel);
|
| H A D | DistributionTemplates.h | 491 void log_normal_kernel(TensorIteratorBase& iter, double mean_, double std_, RNG gen) { in log_normal_kernel() function
507 log_normal_kernel(iter, mean, std, check_generator<RNG>(gen)); in operator()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | DistributionKernels.cpp | 203 static void log_normal_kernel(TensorIteratorBase& iter, double mean, double std, std::optional<Gene… in log_normal_kernel() function
205 templates::cpu::log_normal_kernel(iter, mean, std, generator); in log_normal_kernel()
243 REGISTER_DISPATCH(log_normal_stub, &log_normal_kernel);
|
| H A D | DistributionTemplates.h | 309 void log_normal_kernel(TensorIteratorBase& iter, double mean, double std, RNG generator) { in log_normal_kernel() function
322 log_normal_kernel(iter, mean, std, check_generator<RNG>(gen)); in operator()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | DistributionTemplates.h | 315 template<template<typename> class log_normal_kernel, typename RNG>
320 log_normal_kernel<RNG>()(iter, mean, std, gen); in log_normal_impl_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | cpu_rng_test.cpp | 344 …native::templates::cpu::log_normal_kernel(iter, mean, std, check_generator<TestCPUGenerator>(gen)); in TEST_F()
|