Searched defs:rng_seed (Results 1 – 3 of 3) sorted by relevance
151 se::Stream* stream, int rng_seed) { in RunExecutable()199 se::Stream* stream, int rng_seed) { in ExecuteComputation()218 se::Stream* stream, int rng_seed) { in ExecuteComputation()281 int rng_seed = config_proto.rng_seed(); in DoWork() local362 int rng_seed = config.rng_seed(); in DoWork() local
103 ExecutableRunOptions& ExecutableRunOptions::set_rng_seed(int rng_seed) { in set_rng_seed()108 int ExecutableRunOptions::rng_seed() const { return rng_seed_; } in rng_seed() function in xla::ExecutableRunOptions
126 uint32 rng_seed = 4; field139 uint32 rng_seed = 1; field