Home
last modified time | relevance | path

Searched defs:CPUGeneratorImpl (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dcpu_generator_test.cpp14 TEST(CPUGeneratorImpl, TestGeneratorDynamicCast) { in TEST() argument
21 TEST(CPUGeneratorImpl, TestDefaultGenerator) { in TEST() argument
31 TEST(CPUGeneratorImpl, TestCloning) { in TEST() argument
55 TEST(CPUGeneratorImpl, TestMultithreadingGetEngineOperator) { in TEST() argument
86 TEST(CPUGeneratorImpl, TestGetSetCurrentSeed) { in TEST() argument
104 TEST(CPUGeneratorImpl, TestMultithreadingGetSetCurrentSeed) { in TEST() argument
119 TEST(CPUGeneratorImpl, TestRNGForking) { in TEST() argument
141 TEST(CPUGeneratorImpl, TestPhiloxEngineReproducibility) { in TEST() argument
152 TEST(CPUGeneratorImpl, TestPhiloxEngineOffset1) { in TEST() argument
173 TEST(CPUGeneratorImpl, TestPhiloxEngineOffset2) { in TEST() argument
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DCPUGeneratorImpl.h10 struct TORCH_API CPUGeneratorImpl : public c10::GeneratorImpl { struct
36 at::mt19937 engine_;
37 std::optional<float> next_float_normal_sample_;
38 std::optional<double> next_double_normal_sample_;
H A DCPUGeneratorImpl.cpp81 CPUGeneratorImpl::CPUGeneratorImpl(uint64_t seed_in) in CPUGeneratorImpl() function in at::CPUGeneratorImpl