Home
last modified time | relevance | path

Searched refs:functional_subgraph_node (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dcreate_functional_graphs.cpp79 Node* functional_subgraph_node = in CreateFunctionalGraphsImpl() local
88 if (n->kind() == prim::Constant || n == functional_subgraph_node) { in CreateFunctionalGraphsImpl()
104 isEmptyFunctionalGraph(functional_subgraph_node)) { in CreateFunctionalGraphsImpl()
105 functional_subgraph_node->destroy(); in CreateFunctionalGraphsImpl()
106 functional_subgraph_node = n; in CreateFunctionalGraphsImpl()
111 if (aliasDb_->moveBeforeTopologicallyValid(n, functional_subgraph_node)) { in CreateFunctionalGraphsImpl()
112 SubgraphUtils::mergeNodeIntoSubgraph(n, functional_subgraph_node); in CreateFunctionalGraphsImpl()
114 functional_graph_nodes.emplace_back(functional_subgraph_node); in CreateFunctionalGraphsImpl()
115 functional_subgraph_node = in CreateFunctionalGraphsImpl()
117 SubgraphUtils::mergeNodeIntoSubgraph(n, functional_subgraph_node); in CreateFunctionalGraphsImpl()
[all …]