Home
last modified time | relevance | path

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

/external/armnn/src/armnn/
DSubgraphView.cpp538 void SubgraphView::UpdateSubgraphViewSlotPointers(SubgraphView& patternSubgraph, in UpdateSubgraphViewSlotPointers() argument
543 for (unsigned long idx = 0; idx < patternSubgraph.GetIInputSlots().size(); idx++) in UpdateSubgraphViewSlotPointers()
545 IInputSlot *slot = patternSubgraph.GetIInputSlots()[idx]; in UpdateSubgraphViewSlotPointers()
558 for (unsigned long idx = 0; idx < patternSubgraph.GetIOutputSlots().size(); idx++) in UpdateSubgraphViewSlotPointers()
560 IOutputSlot *slot = patternSubgraph.GetIOutputSlots()[idx]; in UpdateSubgraphViewSlotPointers()
572 void SubgraphView::SubstituteSubgraph(SubgraphView& patternSubgraph, const SubgraphView& substitute… in SubstituteSubgraph() argument
580 auto numPatternInputs = patternSubgraph.GetIInputSlots().size(); in SubstituteSubgraph()
591 auto numPatternOutputs = patternSubgraph.GetIOutputSlots().size(); in SubstituteSubgraph()
622 workingCopyGraph->ReplaceSubgraphConnections(patternSubgraph, substituteSubgraph); in SubstituteSubgraph()
625 UpdateSubgraphViewSlotPointers(patternSubgraph, substituteSubgraph); in SubstituteSubgraph()
[all …]
/external/armnn/src/armnn/test/
DSubgraphViewTests.cpp1924 SubgraphView patternSubgraph(std::move(layers), in ReplaceConstantMultiplicationWithDepthwise() local
1928 subgraph.SubstituteSubgraph(patternSubgraph, depthwiseLayer ); in ReplaceConstantMultiplicationWithDepthwise()