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.cc59 sub_streams_(sub_streams), in HloExecutionProfiler()
73 GetCyclesTaken(&timers_, sub_streams_, stream_, clock_rate_ghz_)); in FinishExecution()
88 GetCyclesTaken(&timers_, sub_streams_, stream_, clock_rate_ghz_)); in FinishHloComputation()
97 GetCyclesTaken(&timers_, sub_streams_, stream_, clock_rate_ghz_)); in FinishHloComputation()
110 profile_->SetCyclesTakenBy(index, GetCyclesTaken(&timers_, sub_streams_, in FinishHloInstruction()
Dhlo_execution_profiler.h82 const std::vector<StreamPool::Ptr>& sub_streams_; variable
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1299 for (size_t index = 0; index < sub_streams_.size();) { in GetOrCreateSubStream()
1300 std::pair<std::unique_ptr<Stream>, bool> &pair = sub_streams_[index]; in GetOrCreateSubStream()
1314 const int64 last = sub_streams_.size() - 1; in GetOrCreateSubStream()
1316 std::swap(pair, sub_streams_[last]); in GetOrCreateSubStream()
1318 bad_streams.push_back(std::move(sub_streams_.back().first)); in GetOrCreateSubStream()
1319 sub_streams_.pop_back(); in GetOrCreateSubStream()
1329 sub_streams_.emplace_back(std::unique_ptr<Stream>{new Stream{parent_}}, in GetOrCreateSubStream()
1331 Stream *sub_stream = sub_streams_.back().first.get(); in GetOrCreateSubStream()
1350 for (int64 index = 0, end = sub_streams_.size(); index < end; ++index) { in ReturnSubStream()
1351 std::pair<std::unique_ptr<Stream>, bool> &pair = sub_streams_[index]; in ReturnSubStream()
[all …]
Dstream.h2078 std::vector<std::pair<std::unique_ptr<Stream>, bool>> sub_streams_