Searched refs:mergeNodeIntoSubgraph (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_subgraph_utils.cpp | 36 SubgraphUtils::mergeNodeIntoSubgraph(*it, subgraph); in TEST()
80 SubgraphUtils::mergeNodeIntoSubgraph(next, graph1); in TEST()
85 SubgraphUtils::mergeNodeIntoSubgraph(graph2->next(), graph2); in TEST()
90 SubgraphUtils::mergeNodeIntoSubgraph(graph2, graph1); in TEST()
93 SubgraphUtils::mergeNodeIntoSubgraph(graph1, graph2); in TEST()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/ |
| H A D | subgraph_utils.cpp | 140 mergeNodeIntoSubgraph(node, mergeTo); in mergeSubgraph()
283 void mergeNodeIntoSubgraph( in mergeNodeIntoSubgraph() function
413 mergeNodeIntoSubgraph(n, subgraph); in createSingletonSubgraph()
422 mergeNodeIntoSubgraph(to_merge, subgraphNode); in mergeNodeIntoSubgraphAndUpdateAliasing()
|
| H A D | subgraph_utils.h | 36 TORCH_API void mergeNodeIntoSubgraph(
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | create_functional_graphs.cpp | 112 SubgraphUtils::mergeNodeIntoSubgraph(n, functional_subgraph_node); in CreateFunctionalGraphsImpl()
117 SubgraphUtils::mergeNodeIntoSubgraph(n, functional_subgraph_node); in CreateFunctionalGraphsImpl()
|
| H A D | symbolic_shape_runtime_fusion.cpp | 470 SubgraphUtils::mergeNodeIntoSubgraph(versioning_if, te_dyn_group); in insertDynamicShapesGuard()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ |
| H A D | graph_rewriter.cpp | 137 llgaHelper_.mergeNodeIntoSubgraph(producer, consumer, aliasDb_); in tryMerge()
|
| H A D | graph_helper.h | 61 void mergeNodeIntoSubgraph(Node* toMerge, Node* subgraphNode, AliasDb& db);
|
| H A D | graph_helper.cpp | 517 void LlgaGraphHelper::mergeNodeIntoSubgraph( in mergeNodeIntoSubgraph() function in torch::jit::fuser::onednn::LlgaGraphHelper
|