Searched refs:execute_backend_ (Results 1 – 3 of 3) sorted by relevance
161 execute_backend_(std::move(execute_backend)) { in Service()163 if (execute_backend_) { in Service()164 if (execute_backend_->device_count() > 0) { in Service()165 CHECK_GE(execute_backend_->device_count(), options_.number_of_replicas()) in Service()170 execute_backend_->platform()->Name()); in Service()171 auto stream_executors = execute_backend_->stream_executors(); in Service()172 for (int i = 0; i < execute_backend_->device_count(); ++i) { in Service()249 if (shaped_buffer->platform() != execute_backend_->platform() || in ResolveAndValidateArguments()256 execute_backend_->device_name(replica_device_ordinal)); in ResolveAndValidateArguments()318 if (execute_backend_ != nullptr && in CreateModuleConfig()[all …]
183 const Backend& backend() const { return *execute_backend_; } in backend()184 Backend* mutable_backend() { return execute_backend_.get(); } in mutable_backend()297 std::unique_ptr<Backend> execute_backend_; variable
176 execute_backend_->stream_executor(build_options.device_ordinal())); in CompileExecutable()179 execute_backend_.get(), executor, in CompileExecutable()