Searched refs:rng_seed (Results 1 – 4 of 4) sorted by relevance
89 ExecutableRunOptions& ExecutableRunOptions::set_rng_seed(int rng_seed) { in set_rng_seed() argument90 rng_seed_ = rng_seed; in set_rng_seed()94 int ExecutableRunOptions::rng_seed() const { return rng_seed_; } in rng_seed() function in xla::ExecutableRunOptions
80 ExecutableRunOptions& set_rng_seed(int rng_seed);81 int rng_seed() const;
196 int rng_seed = config_proto.rng_seed(); in DoWork() local197 if (rng_seed == 0) { in DoWork()198 rng_seed = GetXLARandomSeed(); in DoWork()211 run_options.set_rng_seed(rng_seed); in DoWork()
98 // If non-zero, rng_seed to reset the core with.99 uint32 rng_seed = 4; field