Searched refs:normal_impl_ (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | DistributionTemplates.h | 203 Tensor& normal_impl_(Tensor& self, double mean, double std, std::optional<Generator> gen) { in normal_impl_() function
224 normal_impl_<normal_kernel, RNG>(output, 0, std, gen); in normal_out_impl()
235 normal_impl_<normal_kernel, RNG>(output, 0, 1, gen); in normal_out_impl()
249 normal_impl_<normal_kernel, RNG>(output, 0, 1, gen); in normal_out_impl()
|
| D | Distributions.cpp | 276 …return at::native::templates::normal_impl_<NormalStub, Generator>(self, mean, std, std::move(gen)); in normal_()
280 …return at::native::templates::normal_impl_<NormalMeta, Generator>(self, mean, std, std::move(gen)); in normal_meta_()
|
| /external/pytorch/aten/src/ATen/test/ |
| D | cpu_rng_test.cpp | 62 …return at::native::templates::normal_impl_<native::templates::cpu::NormalKernel, TestCPUGenerator>… in normal_()
|