Searched refs:defaultGenerator (Results 1 – 3 of 3) sorted by relevance
40 const Generator& defaultGenerator(Device device) { in defaultGenerator() function516 auto gen = globalContext().defaultGenerator(c10::DeviceType::CPU); in manual_seed()527 auto cuda_gen = globalContext().defaultGenerator( in manual_seed()540 auto xpu_gen = globalContext().defaultGenerator( in manual_seed()551 auto mps_gen = globalContext().defaultGenerator(c10::DeviceType::MPS); in manual_seed()
631 …return at::globalContext().defaultGenerator(at::Device(c10::DeviceType::PrivateUse1, device_index)… in default_generator()
2338 const auto& defaultGenerator = at::detail::getDefaultCPUGenerator(); in initModule() local2340 (THPGenerator*)THPGenerator_initDefaultGenerator(defaultGenerator); in initModule()