Searched refs:run_options_ (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | service_executable_run_options.h | 37 : 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/ |
D | cluster.cc | 57 run_options_.set_trace_level(RunOptions::NO_TRACE); in DisableDetailedStats() 60 run_options_.set_trace_level(RunOptions::HARDWARE_TRACE); in DisableDetailedStats()
|
D | cluster.h | 140 RunOptions run_options_; variable
|
D | single_machine.cc | 168 RunOptions queue_options = run_options_; in Run() 260 *status = session_->Run(run_options_, feed, {}, fetch, nullptr, in RunWithTimeout()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_compiled_cpu_function.h | 130 run_options_.set_intra_op_thread_pool(pool); in set_thread_pool() 376 xla::ExecutableRunOptions run_options_; variable
|
D | xla_compiled_cpu_function.cc | 55 raw_function_(buffer_table_[result_index_], &run_options_, nullptr, in Run()
|
/external/tensorflow/tensorflow/cc/training/ |
D | queue_runner.cc | 223 run_options_ = run_options; in SetRunArgumentsAndCostGraph() 231 s = sess->Run(run_options_, {}, {}, {op}, nullptr, &metadata); in RealRun()
|
D | queue_runner.h | 135 RunOptions run_options_; variable
|