Searched refs:splitDescriptor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | Split.hpp | 105 armnn::SplitterDescriptor splitDescriptor(numSplits, inputDimSize); in VisitSplitOperator() local
111 splitDescriptor.SetViewSize(j, dimIdx, splitterDimSizes[dimIdx]); in VisitSplitOperator()
113 splitDescriptor.SetViewOriginCoord(j, splitDim, splitterDimSizes[splitDim] * j); in VisitSplitOperator()
129 splitDescriptor); in VisitSplitOperator()
133 armnn::IConnectableLayer* layer = delegateData.m_Network->AddSplitterLayer(splitDescriptor); in VisitSplitOperator()
290 armnn::SplitterDescriptor splitDescriptor(numSplits, inputDimSize); in VisitSplitVOperator() local
304 splitDescriptor.SetViewSize(j, dimIdx, dimSize); in VisitSplitVOperator()
307 splitDescriptor.SetViewOriginCoord(j, splitDim, accumSplit); in VisitSplitVOperator()
324 splitDescriptor); in VisitSplitVOperator()
328 armnn::IConnectableLayer* layer = delegateData.m_Network->AddSplitterLayer(splitDescriptor); in VisitSplitVOperator()
|
| /aosp_15_r20/external/armnn/src/armnn/test/ |
| H A D | SubgraphViewTests.cpp | 2442 SplitterDescriptor splitDescriptor(2,4); variable
2444 Layer* splitLayer = graph.AddLayer<SplitterLayer>(splitDescriptor, "split");
|