Home
last modified time | relevance | path

Searched refs:execute_backend_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc162 execute_backend_(std::move(execute_backend)) { in Service()
164 if (execute_backend_) { in Service()
165 if (execute_backend_->device_count() > 0) { in Service()
166 CHECK_GE(execute_backend_->device_count(), options_.number_of_replicas()) in Service()
172 this, execute_backend_->platform()->Name()); in Service()
173 auto stream_executors = execute_backend_->stream_executors(); in Service()
174 for (int i = 0; i < execute_backend_->device_count(); ++i) { in Service()
262 if (execute_backend_ != nullptr && in CreateModuleConfig()
263 execute_backend_->eigen_intra_op_thread_pool() != nullptr) { in CreateModuleConfig()
264 num_threads = execute_backend_->eigen_intra_op_thread_pool()->NumThreads(); in CreateModuleConfig()
[all …]
Dlocal_service.cc158 execute_backend_->stream_executor(build_options.device_ordinal())); in CompileExecutables()
166 execute_backend_.get(), executor, in CompileExecutables()
183 execute_backend_.get(), {executors}, in CompileExecutables()
Dservice.h183 const Backend& backend() const { return *execute_backend_; } in backend()
184 Backend* mutable_backend() { return execute_backend_.get(); } in mutable_backend()
299 std::unique_ptr<Backend> execute_backend_; variable