Home
last modified time | relevance | path

Searched refs:run_options_ (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dservice_executable_run_options.h37 : run_options_(std::move(run_options)), in run_options_() function
41 const ExecutableRunOptions& run_options() const { return run_options_; } in run_options()
42 ExecutableRunOptions* mutable_run_options() { return &run_options_; } in mutable_run_options()
45 se::Stream* stream() const { return run_options_.stream(); } in stream()
47 return run_options_.allocator(); in allocator()
49 int device_ordinal() const { return run_options_.device_ordinal(); } in device_ordinal()
60 ExecutableRunOptions run_options_;
/external/tensorflow/tensorflow/core/grappler/clusters/
Dcluster.cc57 run_options_.set_trace_level(RunOptions::NO_TRACE); in DisableDetailedStats()
60 run_options_.set_trace_level(RunOptions::HARDWARE_TRACE); in DisableDetailedStats()
Dcluster.h140 RunOptions run_options_; variable
Dsingle_machine.cc168 RunOptions queue_options = run_options_; in Run()
260 *status = session_->Run(run_options_, feed, {}, fetch, nullptr, in RunWithTimeout()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiled_cpu_function.h130 run_options_.set_intra_op_thread_pool(pool); in set_thread_pool()
376 xla::ExecutableRunOptions run_options_; variable
Dxla_compiled_cpu_function.cc55 raw_function_(buffer_table_[result_index_], &run_options_, nullptr, in Run()
/external/tensorflow/tensorflow/cc/training/
Dqueue_runner.cc223 run_options_ = run_options; in SetRunArgumentsAndCostGraph()
231 s = sess->Run(run_options_, {}, {}, {op}, nullptr, &metadata); in RealRun()
Dqueue_runner.h135 RunOptions run_options_; variable