Searched refs:sub_streams (Results 1 – 3 of 3) sorted by relevance
89 std::vector<StreamPool::Ptr> sub_streams; in ExecuteThunks() local90 sub_streams.reserve(thunk_schedule_->StreamCount() - 1); in ExecuteThunks()91 while (sub_streams.size() + 1 < thunk_schedule_->StreamCount()) { in ExecuteThunks()92 sub_streams.emplace_back(); in ExecuteThunks()93 TF_ASSIGN_OR_RETURN(sub_streams.back(), in ExecuteThunks()98 sub_streams, hlo_module_->entry_computation()); in ExecuteThunks()129 (stream_no == 0 ? main_stream : sub_streams[stream_no - 1].get()); in ExecuteThunks()155 main_stream->ThenWaitFor(&sub_streams); in ExecuteThunks()
42 const std::vector<StreamPool::Ptr>& sub_streams, in GetCyclesTaken() argument45 stream->ThenWaitFor(&sub_streams); in GetCyclesTaken()56 const std::vector<StreamPool::Ptr>& sub_streams, in HloExecutionProfiler() argument61 sub_streams_(sub_streams), in HloExecutionProfiler()
43 const std::vector<StreamPool::Ptr>& sub_streams,