HomeSort by relevance Sort by last modified time
    Searched refs:execute_backend_ (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
service.cc 152 execute_backend_(std::move(execute_backend)) {
154 if (execute_backend_) {
155 if (execute_backend_->device_count() > 0) {
156 CHECK_GE(execute_backend_->device_count(), options_.number_of_replicas())
161 execute_backend_->platform()->Name().c_str());
162 for (int i = 0; i < execute_backend_->device_count(); ++i) {
163 if (execute_backend_->device_ordinal_supported(i)) {
165 execute_backend_->stream_executor(i).ValueOrDie();
249 if (shaped_buffer->platform() != execute_backend_->platform() ||
256 execute_backend_->device_name(device_ordinal).c_str())
    [all...]
local_service.cc 142 execute_backend_->stream_executor(build_options.device_ordinal()));
145 execute_backend_.get(), executor,
service.h 245 const Backend& backend() const { return *execute_backend_; }
246 Backend* mutable_backend() { return execute_backend_.get(); }
377 std::unique_ptr<Backend> execute_backend_; member in class:xla::Service

Completed in 191 milliseconds