Searched refs:splitDesc (Results 1 – 5 of 5) sorted by relevance
| /external/armnn/src/backends/neon/test/ |
| D | NeonTensorHandleTests.cpp | 344 SplitterDescriptor splitDesc(numSplit, inputShape.GetNumDimensions()); in BOOST_AUTO_TEST_CASE() local
351 splitDesc.SetViewSize(g, dimIdx, splitterDimSizes[dimIdx]); in BOOST_AUTO_TEST_CASE()
353 splitDesc.SetViewOriginCoord(g, splitAxis, splitterDimSizes[splitAxis] * g); in BOOST_AUTO_TEST_CASE()
358 IConnectableLayer* splitter = net->AddSplitterLayer(splitDesc, "splitter"); in BOOST_AUTO_TEST_CASE()
518 SplitterDescriptor splitDesc(numSplit, inputShape.GetNumDimensions()); in BOOST_AUTO_TEST_CASE() local
525 splitDesc.SetViewSize(g, dimIdx, splitterDimSizes[dimIdx]); in BOOST_AUTO_TEST_CASE()
527 splitDesc.SetViewOriginCoord(g, splitAxis, splitterDimSizes[splitAxis] * g); in BOOST_AUTO_TEST_CASE()
533 IConnectableLayer* splitter = net->AddSplitterLayer(splitDesc, "splitter"); in BOOST_AUTO_TEST_CASE()
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | SplitterEndToEndTestImpl.hpp | 50 SplitterDescriptor splitDesc(numSplit, inputShape.GetNumDimensions()); in CreateSplitterNetwork() local
56 splitDesc.SetViewSize(g, dimIdx, splitterDimSizes[dimIdx]); in CreateSplitterNetwork()
58 splitDesc.SetViewOriginCoord(g, splitAxis, splitterDimSizes[splitAxis] * g); in CreateSplitterNetwork()
61 IConnectableLayer* splitter = net->AddSplitterLayer(splitDesc, "splitter"); in CreateSplitterNetwork()
|
| /external/armnn/src/armnnTfLiteParser/ |
| D | TfLiteParser.cpp | 2548 SplitterDescriptor splitDesc(unpackNum, static_cast<unsigned int>(unpackDimSizes.size())); in ParseUnpack()
2554 splitDesc.SetViewSize(j, dimIdx, unpackDimSizes[dimIdx]); in ParseUnpack()
2556 splitDesc.SetViewOriginCoord(j, unpackAxis, unpackDimSizes[unpackAxis] * j); in ParseUnpack()
2560 IConnectableLayer* layer = m_Network->AddSplitterLayer(splitDesc, layerName.c_str()); in ParseUnpack()
2646 SplitterDescriptor splitDesc(numSplits, inputDimSize); in ParseSplit() local
2652 splitDesc.SetViewSize(j, dimIdx, splitterDimSizes[dimIdx]); in ParseSplit()
2654 splitDesc.SetViewOriginCoord(j, splitDim, splitterDimSizes[splitDim] * j); in ParseSplit()
2658 IConnectableLayer* layer = m_Network->AddSplitterLayer(splitDesc, layerName.c_str()); in ParseSplit()
2782 SplitterDescriptor splitDesc(numSplits, inputDimSize); in ParseSplitV() local
2796 splitDesc.SetViewSize(j, dimIdx, dimSize); in ParseSplitV()
[all …]
|
| /external/armnn/src/armnn/test/ |
| D | SubgraphViewTests.cpp | 1190 ViewsDescriptor splitDesc(numOutputs); in BOOST_AUTO_TEST_CASE() local
1191 graph.AddLayer<SplitterLayer>(splitDesc, name.c_str()); in BOOST_AUTO_TEST_CASE()
|
| /external/armnn/src/armnnTfParser/ |
| D | TfParser.cpp | 2830 SplitterDescriptor splitDesc(num_split); in ParseSplit() local
2836 splitDesc.SetViewSize(g, dimIdx, splitterDimSizes[dimIdx]); in ParseSplit()
2838 splitDesc.SetViewOriginCoord(g, splitDim, splitterDimSizes[splitDim] * g); in ParseSplit()
2841 IConnectableLayer *layer = m_Network->AddSplitterLayer(splitDesc, nodeDef.name().c_str()); in ParseSplit()
|