Searched refs:stream_executors_ (Results 1 – 4 of 4) sorted by relevance
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
130 stream_executors_(stream_executors.begin(), stream_executors.end()) { in Backend()133 platform, stream_executors_); in Backend()134 CHECK(!stream_executors_.empty()) in Backend()169 device_ordinal > stream_executors_.back()->device_ordinal()) { in stream_executor()172 device_ordinal, stream_executors_.back()->device_ordinal()); in stream_executor()174 for (auto* executor : stream_executors_) { in stream_executor()198 return transfer_manager_->ResetDevices(stream_executors_); in ResetDevices()
265 std::vector<StreamExecutor *> stream_executors_;
888 stream_executors_ = {executor}; in StreamExecutorMemoryAllocator()895 stream_executors_(stream_executors.begin(), stream_executors.end()) {} in StreamExecutorMemoryAllocator()934 for (StreamExecutor *se : stream_executors_) { in GetStreamExecutor()