Home
last modified time | relevance | path

Searched refs:set_rng_seed (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.cc89 ExecutableRunOptions& ExecutableRunOptions::set_rng_seed(int rng_seed) { in set_rng_seed() function in xla::ExecutableRunOptions
Dexecutable_run_options.h80 ExecutableRunOptions& set_rng_seed(int rng_seed);
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc364 run_options.set_rng_seed(GetXLARandomSeed()); in Compute()
538 run_options.set_rng_seed(GetXLARandomSeed()); in Compute()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc79 run_options.set_rng_seed(GetXLARandomSeed()); in Run()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc211 run_options.set_rng_seed(rng_seed); in DoWork()