Searched defs:splitAxis (Results 1 – 11 of 11) sorted by relevance
| /external/armnn/src/backends/backendsCommon/test/ |
| D | SplitterEndToEndTestImpl.hpp | 25 unsigned int splitAxis, in CreateSplitterNetwork()
81 unsigned int splitAxis = 0; in Splitter1dEndToEnd() local
109 unsigned int splitAxis = 0; in Splitter2dDim0EndToEnd() local
144 unsigned int splitAxis = 1; in Splitter2dDim1EndToEnd() local
182 unsigned int splitAxis = 0; in Splitter3dDim0EndToEnd() local
230 unsigned int splitAxis = 1; in Splitter3dDim1EndToEnd() local
278 unsigned int splitAxis = 2; in Splitter3dDim2EndToEnd() local
318 unsigned int splitAxis = 0; in Splitter4dDim0EndToEnd() local
398 unsigned int splitAxis = 1; in Splitter4dDim1EndToEnd() local
478 unsigned int splitAxis = 2; in Splitter4dDim2EndToEnd() local
[all …]
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonSplitterWorkload.cpp | 25 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis()
34 unsigned int splitAxis) in NeonSplitterWorkloadValidate()
97 …std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->G… in NeonSplitterWorkload() local
|
| D | NeonUnidirectionalSequenceLstmWorkload.cpp | 131 std::set<unsigned int> splitAxis = ComputeSplitAxis(splitterDesc, timeMajorShapeInput); in NeonUnidirectionalSequenceLstmWorkload() local
|
| D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 129 std::set<unsigned int> splitAxis = ComputeSplitAxis(splitterDesc, timeMajorShapeInput); in NeonUnidirectionalSequenceLstmFloatWorkload() local
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClSplitterWorkload.cpp | 24 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis()
33 unsigned int splitAxis) in ClSplitterWorkloadValidate()
97 …std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->G… in ClSplitterWorkload() local
|
| D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 131 std::set<unsigned int> splitAxis = ComputeSplitAxis(splitterDesc, timeMajorShapeInput); in ClUnidirectionalSequenceLstmFloatWorkload() local
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonTensorHandleTests.cpp | 254 unsigned int splitAxis = 2; variable
460 unsigned int splitAxis = 2; variable
|
| /external/armnn/src/armnn/layers/ |
| D | SplitterLayer.cpp | 66 std::set<unsigned int> splitAxis; in CreateTensors() local
|
| /external/armnn/src/backends/aclCommon/ |
| D | ArmComputeUtils.hpp | 248 std::set<unsigned int> splitAxis; in ComputeSplitAxis() local
|
| /external/armnn/src/backends/neon/ |
| D | NeonLayerSupport.cpp | 1510 std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor, input.GetShape()); in IsSplitterSupported() local
|
| /external/armnn/src/backends/cl/ |
| D | ClLayerSupport.cpp | 1459 std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor, input.GetShape()); in IsSplitterSupported() local
|