Searched refs:GetUnsignedAxis (Results 1 – 8 of 8) sorted by relevance
| /external/armnn/src/backends/cl/workloads/ |
| D | ClArgMinMaxWorkload.cpp | 39 auto unsignedAxis = armnnUtils::GetUnsignedAxis(numDims, descriptor.m_Axis); in ClArgMinMaxWorkloadValidate()
70 auto unsignedAxis = armnnUtils::GetUnsignedAxis(numDims, m_Data.m_Parameters.m_Axis); in ClArgMinMaxWorkload()
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonArgMinMaxWorkload.cpp | 39 auto unsignedAxis = armnnUtils::GetUnsignedAxis(numDims, descriptor.m_Axis); in NeonArgMinMaxWorkloadValidate()
69 auto unsignedAxis = armnnUtils::GetUnsignedAxis(numDims, m_Data.m_Parameters.m_Axis); in NeonArgMinMaxWorkload()
|
| /external/armnn/src/backends/reference/workloads/ |
| D | ArgMinMax.cpp | 21 unsigned int uAxis = armnnUtils::GetUnsignedAxis(inputTensorInfo.GetNumDimensions(), axis); in ArgMinMax()
|
| /external/armnn/include/armnnUtils/ |
| D | TensorUtils.hpp | 59 unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis);
|
| /external/armnn/src/armnn/layers/ |
| D | ArgMinMaxLayer.cpp | 45 auto unsignedAxis = armnnUtils::GetUnsignedAxis(inputNumDimensions, axis); in InferOutputShapes()
|
| /external/armnn/src/armnnUtils/ |
| D | TensorUtils.cpp | 221 unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis) in GetUnsignedAxis() function
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadData.cpp | 688 auto unsignedAxis = armnnUtils::GetUnsignedAxis(inputNumDimensions, m_Parameters.m_Axis); in Validate()
|
| /external/armnn/src/armnnTfLiteParser/ |
| D | TfLiteParser.cpp | 3966 …const unsigned int splitDim = armnnUtils::GetUnsignedAxis(inputTensorInfo.GetNumDimensions(), axis… in ParseSplit()
|