Searched refs:stream_executors_ (Results 1 – 4 of 4) sorted by relevance
31 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()
99 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
155 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()
101 std::vector<se::StreamExecutor*> stream_executors_; variable