Searched defs:snodes (Results 1 – 3 of 3) sorted by relevance
37 def _is_fallback_op_in_snodes(snodes, op): argument185 def _is_fwd_graph(self, snodes): argument201 def _check_fsdp_ops_in_snodes(snodes, is_fwd_graph, expect=True): argument231 snodes, name_to_buf, name_to_fused_node, orig_fn argument
140 std::vector<ShowNode> snodes; in ReInit() local
48 def create_grouped_node_for_allreduce_and_its_deps(snodes): argument