Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_execution_profiler.cc61 sub_streams_(sub_streams), in HloExecutionProfiler()
75 GetCyclesTaken(&timers_, sub_streams_, stream_, clock_rate_ghz_)); in FinishExecution()
90 GetCyclesTaken(&timers_, sub_streams_, stream_, clock_rate_ghz_)); in FinishHloComputation()
106 GetCyclesTaken(&timers_, sub_streams_, stream_, clock_rate_ghz_)); in FinishHloInstruction()
Dhlo_execution_profiler.h75 const std::vector<StreamPool::Ptr>& sub_streams_; variable
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1743 for (int64 index = 0; index < sub_streams_.size();) { in GetOrCreateSubStream()
1744 std::pair<std::unique_ptr<Stream>, bool> &pair = sub_streams_[index]; in GetOrCreateSubStream()
1758 const int64 last = sub_streams_.size() - 1; in GetOrCreateSubStream()
1760 std::swap(pair, sub_streams_[last]); in GetOrCreateSubStream()
1762 sub_streams_.pop_back(); in GetOrCreateSubStream()
1772 sub_streams_.emplace_back(std::unique_ptr<Stream>{new Stream{parent_}}, in GetOrCreateSubStream()
1774 Stream *sub_stream = sub_streams_.back().first.get(); in GetOrCreateSubStream()
1789 for (int64 index = 0; index < sub_streams_.size(); ++index) { in ReturnSubStream()
1790 std::pair<std::unique_ptr<Stream>, bool> &pair = sub_streams_[index]; in ReturnSubStream()
1806 const int64 last = sub_streams_.size() - 1; in ReturnSubStream()
[all …]
Dstream.h2015 std::vector<std::pair<std::unique_ptr<Stream>, bool>> sub_streams_