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()
46 DeviceMemoryAllocator* allocator() const { return run_options_.allocator(); } in allocator()
47 int device_ordinal() const { return run_options_.device_ordinal(); } in device_ordinal()
58 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 | 135 RunOptions run_options_; variable
|
| D | single_machine.cc | 169 RunOptions queue_options = run_options_; in Run()
263 *status = session_->Run(run_options_, feed, {}, fetch, nullptr, in RunWithTimeout()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiled_cpu_function.h | 125 run_options_.set_intra_op_thread_pool(pool); in set_thread_pool()
343 xla::ExecutableRunOptions run_options_; variable
|
| D | xla_compiled_cpu_function.cc | 52 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
|