| /external/armnn/src/backends/backendsCommon/test/ |
| D | SplitterEndToEndTestImpl.hpp | 25 unsigned int splitAxis, in CreateSplitterNetwork() argument
44 if (splitterDimSizes[splitAxis] % numSplit != 0) in CreateSplitterNetwork()
48 splitterDimSizes[splitAxis] /= numSplit; in CreateSplitterNetwork()
58 splitDesc.SetViewOriginCoord(g, splitAxis, splitterDimSizes[splitAxis] * g); in CreateSplitterNetwork()
81 unsigned int splitAxis = 0; in Splitter1dEndToEnd() local
87 … INetworkPtr net = CreateSplitterNetwork<ArmnnType>(inputShape, outputShapes, splitAxis, numSplit); in Splitter1dEndToEnd()
109 unsigned int splitAxis = 0; in Splitter2dDim0EndToEnd() local
115 … INetworkPtr net = CreateSplitterNetwork<ArmnnType>(inputShape, outputShapes, splitAxis, numSplit); in Splitter2dDim0EndToEnd()
144 unsigned int splitAxis = 1; in Splitter2dDim1EndToEnd() local
150 … INetworkPtr net = CreateSplitterNetwork<ArmnnType>(inputShape, outputShapes, splitAxis, numSplit); in Splitter2dDim1EndToEnd()
[all …]
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonSplitterWorkload.cpp | 25 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis() argument
27 return (numDimensions - splitAxis) - 1; in CalcAclAxis()
34 unsigned int splitAxis) in NeonSplitterWorkloadValidate() argument
52 unsigned int aclAxis = CalcAclAxis(input.GetNumDimensions(), splitAxis); in NeonSplitterWorkloadValidate()
91 …std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->G… in NeonSplitterWorkload() local
92 if (splitAxis.size() != 1) in NeonSplitterWorkload()
97 …unsigned int aclAxis = CalcAclAxis(descriptor.m_Parameters.GetNumDimensions(), *splitAxis.begin()); in NeonSplitterWorkload()
|
| D | NeonSplitterWorkload.hpp | 20 unsigned int splitAxis);
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClSplitterWorkload.cpp | 25 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis() argument
27 return (numDimensions - splitAxis) - 1; in CalcAclAxis()
34 unsigned int splitAxis) in ClSplitterWorkloadValidate() argument
52 unsigned int aclAxis = CalcAclAxis(input.GetNumDimensions(), splitAxis); in ClSplitterWorkloadValidate()
91 …std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->G… in ClSplitterWorkload() local
92 if (splitAxis.size() != 1) in ClSplitterWorkload()
97 …unsigned int aclAxis = CalcAclAxis(descriptor.m_Parameters.GetNumDimensions(), *splitAxis.begin()); in ClSplitterWorkload()
|
| D | ClSplitterWorkload.hpp | 20 unsigned int splitAxis);
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonTensorHandleTests.cpp | 253 unsigned int splitAxis = 2; in BOOST_AUTO_TEST_CASE() local
337 if (splitterDimSizes[splitAxis] % numSplit != 0) in BOOST_AUTO_TEST_CASE()
342 splitterDimSizes[splitAxis] /= numSplit; in BOOST_AUTO_TEST_CASE()
353 splitDesc.SetViewOriginCoord(g, splitAxis, splitterDimSizes[splitAxis] * g); in BOOST_AUTO_TEST_CASE()
457 unsigned int splitAxis = 2; in BOOST_AUTO_TEST_CASE() local
511 if (splitterDimSizes[splitAxis] % numSplit != 0) in BOOST_AUTO_TEST_CASE()
516 splitterDimSizes[splitAxis] /= numSplit; in BOOST_AUTO_TEST_CASE()
527 splitDesc.SetViewOriginCoord(g, splitAxis, splitterDimSizes[splitAxis] * g); in BOOST_AUTO_TEST_CASE()
|
| /external/armnn/src/backends/aclCommon/ |
| D | ArmComputeUtils.hpp | 214 std::set<unsigned int> splitAxis; in ComputeSplitAxis() local
222 splitAxis.insert(dimIdx); in ComputeSplitAxis()
226 return splitAxis; in ComputeSplitAxis()
|
| /external/armnn/src/armnn/layers/ |
| D | SplitterLayer.cpp | 66 std::set<unsigned int> splitAxis; in CreateTensors() local
74 splitAxis.insert(dimIdx); in CreateTensors()
78 return splitAxis; in CreateTensors()
|
| /external/armnn/src/backends/neon/ |
| D | NeonLayerSupport.cpp | 899 std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor, input.GetShape()); in IsSplitterSupported() local
900 if (descriptor.GetNumDimensions() > 2 && splitAxis.size() == 1 && in IsSplitterSupported()
901 *splitAxis.begin() == descriptor.GetNumDimensions() - 1 ) in IsSplitterSupported()
907 *splitAxis.begin()); in IsSplitterSupported()
|
| /external/armnn/src/backends/cl/ |
| D | ClLayerSupport.cpp | 902 std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor, input.GetShape()); in IsSplitterSupported() local
903 if (descriptor.GetNumDimensions() > 2 && splitAxis.size() == 1 && in IsSplitterSupported()
904 *splitAxis.begin() == descriptor.GetNumDimensions() - 1 ) in IsSplitterSupported()
910 *splitAxis.begin()); in IsSplitterSupported()
|