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()
76 int64 rng_state = 0; in DoUncachedGemmAutotune() local124 int64 rng_state = 0; in DoUncachedGemmAutotune() local
253 int64* rng_state) { in InitializeTypedBuffer()305 int64* rng_state, se::DeviceMemoryBase buffer) { in InitializeBuffer()
387 int64 rng_state = 0; in PickBestAlgorithmNoCacheCuda() local
2455 HloInstruction* rng_state) { in HandleRngGetAndUpdateState()
2758 Status IrEmitter::HandleRngGetAndUpdateState(HloInstruction* rng_state) { in HandleRngGetAndUpdateState()