Home
last modified time | relevance | path

Searched refs:createCPUGenerator (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dcpu_generator_test.cpp16 auto foo = at::detail::createCPUGenerator(); in TEST()
36 auto gen1 = at::detail::createCPUGenerator(); in TEST()
42 auto gen2 = at::detail::createCPUGenerator(); in TEST()
61 auto gen1 = at::detail::createCPUGenerator(); in TEST()
63 auto gen2 = at::detail::createCPUGenerator(); in TEST()
125 auto current_gen = at::detail::createCPUGenerator(); in TEST()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DCPUGeneratorImpl.cpp57 static auto default_gen_cpu = createCPUGenerator(c10::detail::getNonDeterministicRandom()); in getDefaultCPUGenerator()
64 Generator createCPUGenerator(uint64_t seed_val) { in createCPUGenerator() function
H A DCPUGeneratorImpl.h45 createCPUGenerator(uint64_t seed_val = default_rng_seed_val);
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_ops_utils.cpp409 return at::detail::createCPUGenerator(seed.value()); in make_generator_for_device()
411 return at::detail::createCPUGenerator(); in make_generator_for_device()