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

  /external/tensorflow/tensorflow/compiler/xla/service/
device_memory_allocator.cc 31 stream_executors_(stream_executors.begin(), stream_executors.end()) {}
68 if (device_ordinal >= stream_executors_.size()) {
71 stream_executors_.size());
73 if (stream_executors_[device_ordinal] == nullptr) {
77 return stream_executors_[device_ordinal];
backend.h 99 int device_count() const { return stream_executors_.size(); }
107 return stream_executors_;
117 CHECK(!stream_executors_.empty());
118 return stream_executors_[0];
137 stream_executors_[device_ordinal] != nullptr);
172 // Vector of stream executors. stream_executors_[0] is the default executor.
173 std::vector<se::StreamExecutor*> stream_executors_; member in namespace:xla
backend.cc 155 stream_executors_.push_back(exec);
161 CHECK(!stream_executors_.empty())
196 device_ordinal > stream_executors_.back()->device_ordinal()) {
199 device_ordinal, stream_executors_.back()->device_ordinal());
201 for (auto* executor : stream_executors_) {
225 return transfer_manager_->ResetDevices(stream_executors_);
device_memory_allocator.h 101 std::vector<se::StreamExecutor*> stream_executors_; member in class:xla::StreamExecutorMemoryAllocator

Completed in 57 milliseconds