Searched refs:ComputeSplitAxis (Results 1 – 9 of 9) sorted by relevance
| /external/armnn/src/armnn/layers/ |
| D | SplitterLayer.cpp | 62 auto ComputeSplitAxis = [&](const armnn::SplitterDescriptor& desc, const TensorShape& input) in CreateTensors() local
81 std::set<unsigned int> axis = ComputeSplitAxis(m_Param, parentInfo.GetShape()); in CreateTensors()
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonSplitterWorkload.cpp | 97 …std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->G… in NeonSplitterWorkload()
|
| D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 129 std::set<unsigned int> splitAxis = ComputeSplitAxis(splitterDesc, timeMajorShapeInput); in NeonUnidirectionalSequenceLstmFloatWorkload()
|
| D | NeonUnidirectionalSequenceLstmWorkload.cpp | 131 std::set<unsigned int> splitAxis = ComputeSplitAxis(splitterDesc, timeMajorShapeInput); in NeonUnidirectionalSequenceLstmWorkload()
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClSplitterWorkload.cpp | 97 …std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->G… in ClSplitterWorkload()
|
| D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 131 std::set<unsigned int> splitAxis = ComputeSplitAxis(splitterDesc, timeMajorShapeInput); in ClUnidirectionalSequenceLstmFloatWorkload()
|
| /external/armnn/src/backends/aclCommon/ |
| D | ArmComputeUtils.hpp | 244 inline std::set<unsigned int> ComputeSplitAxis(const armnn::SplitterDescriptor& desc, const TensorS… in ComputeSplitAxis() function
|
| /external/armnn/src/backends/cl/ |
| D | ClLayerSupport.cpp | 1459 std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor, input.GetShape()); in IsSplitterSupported()
|
| /external/armnn/src/backends/neon/ |
| D | NeonLayerSupport.cpp | 1510 std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor, input.GetShape()); in IsSplitterSupported()
|