Searched refs:random_from_to_kernel (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DistributionRandomKernel.cu | 8 void random_from_to_kernel(TensorIteratorBase& iter, uint64_t range, int64_t base, std::optional<Ge… in random_from_to_kernel() function
10 at::native::templates::cuda::random_from_to_kernel(iter, range, base, gen); in random_from_to_kernel()
23 REGISTER_DISPATCH(random_from_to_stub, &random_from_to_kernel);
|
| H A D | DistributionTemplates.h | 287 void random_from_to_kernel(TensorIteratorBase& iter, uint64_t range, int64_t base, RNG gen) { in random_from_to_kernel() function
355 random_from_to_kernel(iter, range, base, check_generator<RNG>(gen)); in operator()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | DistributionKernels.cpp | 218 static void random_from_to_kernel(TensorIteratorBase& iter, uint64_t range, int64_t base, std::opti… in random_from_to_kernel() function
220 templates::cpu::random_from_to_kernel(iter, range, base, generator); in random_from_to_kernel()
246 REGISTER_DISPATCH(random_from_to_stub, &random_from_to_kernel);
|
| H A D | DistributionTemplates.h | 26 void random_from_to_kernel(TensorIteratorBase& iter, uint64_t range, int64_t base, RNG generator) { in random_from_to_kernel() function
60 random_from_to_kernel(iter, range, base, check_generator<RNG>(gen)); in operator()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | DistributionTemplates.h | 134 template<template<typename> class random_from_to_kernel, typename RNG>
152 random_from_to_kernel<RNG>()(iter, range, from, generator); in random_from_to_impl()
177 random_from_to_kernel<RNG>()(iter, range, from, generator); in random_from_to_impl()
182 random_from_to_kernel<RNG>()(iter, generator); in random_from_to_impl()
|