Searched refs:uniform_kernel (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | DistributionUniform.cu | 8 void uniform_kernel(TensorIteratorBase& iter, double from, double to, std::optional<Generator> gen)… in uniform_kernel() function
10 templates::cuda::uniform_kernel(iter, from, to, generator); in uniform_kernel()
13 REGISTER_DISPATCH(uniform_stub, &uniform_kernel);
|
| D | DistributionTemplates.h | 458 void uniform_kernel(TensorIteratorBase& iter, double from_, double to_, RNG gen) { in uniform_kernel() function
484 uniform_kernel(iter, from, to, check_generator<RNG>(gen)); in operator()
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | DistributionKernels.cpp | 208 void uniform_kernel(TensorIteratorBase& iter, double from, double to, std::optional<Generator> gen)… in uniform_kernel() function
210 templates::cpu::uniform_kernel(iter, from, to, generator); in uniform_kernel()
245 REGISTER_DISPATCH(uniform_stub, &uniform_kernel);
|
| D | DistributionTemplates.h | 267 void uniform_kernel(TensorIteratorBase& iter, double from_, double to_, RNG generator) { in uniform_kernel() function
282 uniform_kernel(iter, from, to, check_generator<RNG>(gen)); in operator()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | DistributionTemplates.h | 285 template<template<typename> class uniform_kernel, typename RNG>
290 uniform_impl_<uniform_kernel, RNG>(float_tensor, from, to, generator); in uniform_impl_()
308 uniform_kernel<RNG>()(iter, from, to, generator); in uniform_impl_()
|
| /external/pytorch/aten/src/ATen/test/ |
| D | cpu_rng_test.cpp | 310 native::templates::cpu::uniform_kernel(iter, from, to, check_generator<TestCPUGenerator>(gen)); in TEST_F()
|