Searched refs:leaf_streams (Results 1 – 2 of 2) sorted by relevance
124 std::unordered_set<c10::Stream> leaf_streams; member
729 if (!leaf_streams.empty()) { in exec_post_processing()730 for (const auto& leaf_stream : leaf_streams) { in exec_post_processing()1029 graph_task->leaf_streams.emplace(*opt_parent_stream); in evaluate_function()1052 graph_task->leaf_streams.emplace(*opt_parent_stream); in evaluate_function()