Searched refs:set_rng_seed (Results 1 – 7 of 7) sorted by relevance
116 ExecutableRunOptions& ExecutableRunOptions::set_rng_seed(int rng_seed) { in set_rng_seed() function in xla::ExecutableRunOptions
130 ExecutableRunOptions& set_rng_seed(int rng_seed);
85 run_options.set_rng_seed(GetXLARandomSeed()); in Run()
275 run_options.set_rng_seed(GetXLARandomSeed()); in Compute()511 run_options.set_rng_seed(GetXLARandomSeed()); in Compute()
443 run_options.set_rng_seed(rng_seed); in TPUExecute()
315 run_options.set_rng_seed(rng_seed); in RunExecutable()
1756 run_options.set_rng_seed(device_state->GetNewPrngSeed()); in EnqueueExecution()