| /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()
97 …std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->G… in NeonSplitterWorkload() local
98 if (splitAxis.size() != 1) in NeonSplitterWorkload()
103 …unsigned int aclAxis = CalcAclAxis(descriptor.m_Parameters.GetNumDimensions(), *splitAxis.begin()); in NeonSplitterWorkload()
|
| D | NeonSplitterWorkload.hpp | 20 unsigned int splitAxis);
|
| D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 129 std::set<unsigned int> splitAxis = ComputeSplitAxis(splitterDesc, timeMajorShapeInput); in NeonUnidirectionalSequenceLstmFloatWorkload() local
133 *splitAxis.begin()); in NeonUnidirectionalSequenceLstmFloatWorkload()
|
| D | NeonUnidirectionalSequenceLstmWorkload.cpp | 131 std::set<unsigned int> splitAxis = ComputeSplitAxis(splitterDesc, timeMajorShapeInput); in NeonUnidirectionalSequenceLstmWorkload() local
135 *splitAxis.begin()); in NeonUnidirectionalSequenceLstmWorkload()
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClSplitterWorkload.cpp | 24 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis() argument
26 return (numDimensions - splitAxis) - 1; in CalcAclAxis()
33 unsigned int splitAxis) in ClSplitterWorkloadValidate() argument
51 unsigned int aclAxis = CalcAclAxis(input.GetNumDimensions(), splitAxis); in ClSplitterWorkloadValidate()
97 …std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->G… in ClSplitterWorkload() local
98 if (splitAxis.size() != 1) in ClSplitterWorkload()
103 …unsigned int aclAxis = CalcAclAxis(descriptor.m_Parameters.GetNumDimensions(), *splitAxis.begin()); in ClSplitterWorkload()
|
| D | ClSplitterWorkload.hpp | 21 unsigned int splitAxis);
|
| D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 131 std::set<unsigned int> splitAxis = ComputeSplitAxis(splitterDesc, timeMajorShapeInput); in ClUnidirectionalSequenceLstmFloatWorkload() local
134 … unsigned int aclAxisSplit = CalcAclAxis(splitterDesc.GetNumDimensions(), *splitAxis.begin()); in ClUnidirectionalSequenceLstmFloatWorkload()
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonTensorHandleTests.cpp | 254 unsigned int splitAxis = 2; variable
338 if (splitterDimSizes[splitAxis] % numSplit != 0)
343 splitterDimSizes[splitAxis] /= numSplit;
354 splitDesc.SetViewOriginCoord(g, splitAxis, splitterDimSizes[splitAxis] * g);
460 unsigned int splitAxis = 2; variable
514 if (splitterDimSizes[splitAxis] % numSplit != 0)
519 splitterDimSizes[splitAxis] /= numSplit;
530 splitDesc.SetViewOriginCoord(g, splitAxis, splitterDimSizes[splitAxis] * g);
|
| /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/aclCommon/ |
| D | ArmComputeUtils.hpp | 248 std::set<unsigned int> splitAxis; in ComputeSplitAxis() local
256 splitAxis.insert(dimIdx); in ComputeSplitAxis()
260 return splitAxis; in ComputeSplitAxis()
|
| /external/armnn/src/backends/cl/ |
| D | ClLayerSupport.cpp | 1459 std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor, input.GetShape()); in IsSplitterSupported() local
1460 if (descriptor.GetNumDimensions() > 2 && splitAxis.size() == 1 && in IsSplitterSupported()
1461 *splitAxis.begin() == descriptor.GetNumDimensions() - 1 ) in IsSplitterSupported()
1467 *splitAxis.begin()); in IsSplitterSupported()
|
| /external/armnn/src/backends/neon/ |
| D | NeonLayerSupport.cpp | 1510 std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor, input.GetShape()); in IsSplitterSupported() local
1511 if (descriptor.GetNumDimensions() > 2 && splitAxis.size() == 1 && in IsSplitterSupported()
1512 *splitAxis.begin() == descriptor.GetNumDimensions() - 1 ) in IsSplitterSupported()
1518 *splitAxis.begin()); in IsSplitterSupported()
|