Searched defs:rng_state (Results 1 – 7 of 7) sorted by relevance
33 double generate_uniform_random_number(uint32_t* rng_state) { in generate_uniform_random_number()39 double generate_uniform_random_number_between(uint32_t* rng_state, double a, in generate_uniform_random_number_between()
246 int64* rng_state) { in InitializeTypedBuffer()298 int64* rng_state, se::DeviceMemoryBase buffer) { in InitializeBuffer()
89 int64 rng_state = 0; in DoUncachedGemmAutotune() local267 int64 rng_state = 0; in RunOnInstruction() local
374 int64 rng_state = 0; in PickBestAlgorithmNoCacheCuda() local
819 HloInstruction* rng_state) { in HandleRngGetAndUpdateState()
688 uint64_t rng_state[2]; in construct_block_size_descriptor_2d() local854 uint64_t rng_state[2]; in construct_block_size_descriptor_3d() local
2901 Status IrEmitter::HandleRngGetAndUpdateState(HloInstruction* rng_state) { in HandleRngGetAndUpdateState()