Searched refs:sub_streams (Results 1 – 3 of 3) sorted by relevance
41 const std::vector<StreamPool::Ptr>& sub_streams, in GetCyclesTaken() argument44 stream->ThenWaitFor(&sub_streams); in GetCyclesTaken()55 const std::vector<StreamPool::Ptr>& sub_streams, size_t index) in HloExecutionProfiler() argument59 sub_streams_(sub_streams), in HloExecutionProfiler()
147 std::vector<StreamPool::Ptr> sub_streams; in ExecuteThunks() local148 sub_streams.reserve(thunk_schedule_->StreamCount() - 1); in ExecuteThunks()149 while (sub_streams.size() + 1 < thunk_schedule_->StreamCount()) { in ExecuteThunks()150 sub_streams.emplace_back(); in ExecuteThunks()151 TF_ASSIGN_OR_RETURN(sub_streams.back(), in ExecuteThunks()155 sub_streams.back()->ThenWaitFor(main_stream); in ExecuteThunks()159 sub_streams, entry_computation_profile_index_); in ExecuteThunks()176 (stream_no == 0 ? main_stream : sub_streams[stream_no - 1].get()); in ExecuteThunks()208 main_stream->ThenWaitFor(&sub_streams); in ExecuteThunks()
42 const std::vector<StreamPool::Ptr>& sub_streams,