Searched refs:SubgraphUtils (Results 1 – 15 of 15) sorted by relevance
| /external/pytorch/test/cpp/jit/ |
| D | test_subgraph_utils.cpp | 29 subgraph = SubgraphUtils::createSingletonSubgraph( in TEST()
36 SubgraphUtils::mergeNodeIntoSubgraph(*it, subgraph); in TEST()
43 SubgraphUtils::unmergeSubgraph(subgraph); in TEST()
73 Node* graph1 = SubgraphUtils::createSingletonSubgraph( in TEST()
80 SubgraphUtils::mergeNodeIntoSubgraph(next, graph1); in TEST()
82 Node* graph2 = SubgraphUtils::createSingletonSubgraph( in TEST()
85 SubgraphUtils::mergeNodeIntoSubgraph(graph2->next(), graph2); in TEST()
90 SubgraphUtils::mergeNodeIntoSubgraph(graph2, graph1); in TEST()
93 SubgraphUtils::mergeNodeIntoSubgraph(graph1, graph2); in TEST()
110 SubgraphUtils::unmergeSubgraph(subgraph); in TEST()
[all …]
|
| D | test_alias_analysis.cpp | 1503 SubgraphUtils::createSingletonSubgraph(node, prim::FunctionalGraph); in TEST()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | create_functional_graphs.cpp | 58 auto subgraph = SubgraphUtils::getSubgraph(n); in inlineIfTooSmall()
62 SubgraphUtils::unmergeSubgraph(n); in inlineIfTooSmall()
112 SubgraphUtils::mergeNodeIntoSubgraph(n, functional_subgraph_node); in CreateFunctionalGraphsImpl()
117 SubgraphUtils::mergeNodeIntoSubgraph(n, functional_subgraph_node); in CreateFunctionalGraphsImpl()
205 SubgraphUtils::unmergeSubgraph(n); in InlineFunctionalGraphs()
|
| D | create_autodiff_subgraphs.cpp | 137 any_changed |= SubgraphUtils::unmergeAliasedOutputs(n); in unfuseAliasedOutputs()
138 any_changed |= SubgraphUtils::unmergeOutputsAlisingInputs(n); in unfuseAliasedOutputs()
201 auto subgraph = SubgraphUtils::getSubgraph(n); in inlineIfTooSmall()
211 SubgraphUtils::unmergeSubgraph(n); in inlineIfTooSmall()
264 consumer = SubgraphUtils::createSingletonSubgraphAndUpdateAliasing( in scanNode()
291 SubgraphUtils::mergeNodeIntoSubgraphAndUpdateAliasing( in tryMerge()
|
| D | symbolic_shape_runtime_fusion.cpp | 258 auto tensorexpr_graph = SubgraphUtils::getSubgraph(tensorexpr_graph_node); in GenerateGuard()
319 SubgraphUtils::unmergeSubgraph(n); in inlineFallbackGraphAndAddSRCopyOutOp()
349 auto subgraph = SubgraphUtils::getSubgraph(guarded_node); in insertDynamicShapesGuard()
468 auto te_dyn_group = SubgraphUtils::createSingletonSubgraph( in insertDynamicShapesGuard()
470 SubgraphUtils::mergeNodeIntoSubgraph(versioning_if, te_dyn_group); in insertDynamicShapesGuard()
472 SubgraphUtils::getSubgraph(te_dyn_group)); in insertDynamicShapesGuard()
|
| D | inline_autodiff_subgraphs.cpp | 71 SubgraphUtils::unmergeSubgraph(node); in scanNode()
|
| D | tensorexpr_fuser.cpp | 304 auto subgraph = SubgraphUtils::getSubgraph(guarded_node); in insertTypeGuard()
576 return SubgraphUtils::createSingletonSubgraphAndUpdateAliasing( in getOrCreateTensorExprSubgraph()
746 auto subgraph = SubgraphUtils::getSubgraph(n); in inlineIfTooSmall()
753 SubgraphUtils::unmergeSubgraph(n); in inlineIfTooSmall()
807 SubgraphUtils::mergeNodeIntoSubgraphAndUpdateAliasing( in tryMerge()
1286 SubgraphUtils::unmergeSubgraph(fusion_group); in generalizeFusionGroups()
|
| D | frozen_ops_to_mkldnn.cpp | 747 auto subgraph = SubgraphUtils::getSubgraph(subgraph_node); in ComputeSubgraphInMKLDNN()
1050 InplaceMKLDNNSubgraph(SubgraphUtils::getSubgraph(curNode)); in computeSubgraphsInMKLDNN()
1051 SubgraphUtils::unmergeSubgraph(curNode); in computeSubgraphsInMKLDNN()
1074 producer = SubgraphUtils::createSingletonSubgraphAndUpdateAliasing( in scanNode()
1110 SubgraphUtils::mergeNodeIntoSubgraphAndUpdateAliasing( in tryMerge()
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | fusion.cpp | 144 return SubgraphUtils::createSingletonSubgraphAndUpdateAliasing( in getOrCreateStaticSubgraph()
202 SubgraphUtils::mergeNodeIntoSubgraphAndUpdateAliasing( in tryMerge()
247 auto subgraph = SubgraphUtils::getSubgraph(n); in inlineIfTooSmall()
252 SubgraphUtils::unmergeSubgraph(n); in inlineIfTooSmall()
330 SubgraphUtils::unmergeSubgraph(n); in inlineFallbackGraphs()
|
| /external/pytorch/torch/csrc/jit/passes/utils/ |
| D | subgraph_utils.h | 14 namespace SubgraphUtils {
|
| D | subgraph_utils.cpp | 14 namespace SubgraphUtils { namespace
451 SubgraphUtils::unmergeNode(*it, subgraphNode); in unmergeOutputsAlisingInputs()
|
| /external/pytorch/torch/csrc/jit/codegen/onednn/ |
| D | graph_helper.cpp | 511 auto group = SubgraphUtils::createSingletonSubgraphAndUpdateAliasing( in createSingletonSubgraph()
541 SubgraphUtils::mergeNodeIntoSubgraphAndUpdateAliasing( in mergeNodeIntoSubgraph()
561 SubgraphUtils::unmergeSubgraph(subgraphNode); in unmergeIfAnyNodeIsMissing()
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | kernel.h | 143 SubgraphUtils::generateNameForGraph(subgraph), in TensorExprKernel()
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | profiling_graph_executor_impl.cpp | 324 SubgraphUtils::unmergeSubgraph(dnode); in guardDifferentiableGraph()
|
| /external/armnn/ |
| D | LICENSE.spdx | 10798 FileName: ./src/backends/backendsCommon/SubgraphUtils.hpp
|