Home
last modified time | relevance | path

Searched refs:stream_executors_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddevice_memory_allocator.cc31 stream_executors_(stream_executors.begin(), stream_executors.end()) {} in StreamExecutorMemoryAllocator()
68 if (device_ordinal >= stream_executors_.size()) { in GetStreamExecutor()
71 stream_executors_.size()); in GetStreamExecutor()
73 if (stream_executors_[device_ordinal] == nullptr) { in GetStreamExecutor()
77 return stream_executors_[device_ordinal]; in GetStreamExecutor()
Dbackend.h99 int device_count() const { return stream_executors_.size(); } in device_count()
107 return stream_executors_; in stream_executors()
117 CHECK(!stream_executors_.empty()); in default_stream_executor()
118 return stream_executors_[0]; in default_stream_executor()
137 stream_executors_[device_ordinal] != nullptr); in device_ordinal_supported()
173 std::vector<se::StreamExecutor*> stream_executors_; variable
Dbackend.cc155 stream_executors_.push_back(exec); in Backend()
161 CHECK(!stream_executors_.empty()) in Backend()
196 device_ordinal > stream_executors_.back()->device_ordinal()) { in stream_executor()
199 device_ordinal, stream_executors_.back()->device_ordinal()); in stream_executor()
201 for (auto* executor : stream_executors_) { in stream_executor()
225 return transfer_manager_->ResetDevices(stream_executors_); in ResetDevices()
Ddevice_memory_allocator.h101 std::vector<se::StreamExecutor*> stream_executors_; variable