| /external/tensorflow/tensorflow/compiler/xla/service/ |
| service_executable_run_options.h | 39 : run_options_(std::move(run_options)),
44 const ExecutableRunOptions& run_options() const { return run_options_; }
45 ExecutableRunOptions* mutable_run_options() { return &run_options_; }
48 perftools::gputools::Stream* stream() const { return run_options_.stream(); }
49 DeviceMemoryAllocator* allocator() const { return run_options_.allocator(); }
50 int device_ordinal() const { return run_options_.device_ordinal(); }
67 ExecutableRunOptions run_options_; member in class:xla::ServiceExecutableRunOptions
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_compiled_cpu_function.h | 111 run_options_.set_intra_op_thread_pool(pool);
117 raw_function_(temps_[result_index_], &run_options_,
235 xla::ExecutableRunOptions run_options_; member in class:tensorflow::XlaCompiledCpuFunction
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| cluster.cc | 44 run_options_.set_trace_level(RunOptions::NO_TRACE);
47 run_options_.set_trace_level(RunOptions::HARDWARE_TRACE);
|
| cluster.h | 124 RunOptions run_options_; member in class:tensorflow::grappler::Cluster
|
| single_machine.cc | 156 RunOptions queue_options = run_options_;
250 *status = session_->Run(run_options_, feed, {}, fetch, nullptr,
|
| /external/tensorflow/tensorflow/cc/training/ |
| queue_runner.h | 135 RunOptions run_options_; member in class:tensorflow::QueueRunner
|
| queue_runner.cc | 223 run_options_ = run_options;
231 s = sess->Run(run_options_, {}, {}, {op}, nullptr, &metadata);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| parallel_cpu_executable.cc | 112 run_options_(run_options),
117 thread_pool_(CHECK_NOTNULL(run_options_->xla_intra_op_thread_pool())),
147 const ServiceExecutableRunOptions* run_options_; member in class:xla::cpu::__anon38883::Executor
263 const auto* exec_run_options = &run_options_->run_options();
|
