| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClSoftmaxWorkload.cpp | 24 int aclAxis = ComputeAclAxis(descriptor.m_Axis, input); in ClSoftmaxWorkloadValidate() local
46 int aclAxis = ComputeAclAxis(m_Data.m_Parameters.m_Axis, info.m_InputTensorInfos[0]); in ClSoftmaxWorkload() local
|
| H A D | ClLogSoftmaxWorkload.cpp | 24 int aclAxis = ComputeAclAxis(descriptor.m_Axis, input); in ClLogSoftmaxWorkloadValidate() local
46 int aclAxis = ComputeAclAxis(m_Data.m_Parameters.m_Axis, info.m_InputTensorInfos[0]); in ClLogSoftmaxWorkload() local
|
| H A D | ClGatherWorkload.cpp | 24 int aclAxis = ComputeAclAxis(descriptor.m_Axis, input); in ClGatherWorkloadValidate() local
46 int aclAxis = ComputeAclAxis(descriptor.m_Parameters.m_Axis, info.m_InputTensorInfos[0]); in ClGatherWorkload() local
|
| H A D | ClArgMinMaxWorkload.cpp | 40 int aclAxis = armnn::numeric_cast<int>(CalcAclAxis(numDims, unsignedAxis)); in ClArgMinMaxWorkloadValidate() local
71 int aclAxis = armnn::numeric_cast<int>(CalcAclAxis(numDims, unsignedAxis)); in ClArgMinMaxWorkload() local
|
| H A D | ClStackWorkload.cpp | 42 int aclAxis = CalcAxis(descriptor.m_Axis, descriptor.m_InputShape.GetNumDimensions()); in ClStackWorkloadValidate() local
68 …int aclAxis = CalcAxis(descriptor.m_Parameters.m_Axis, descriptor.m_Parameters.m_InputShape.GetNum… in ClStackWorkload() local
|
| H A D | ClConcatWorkload.cpp | 44 size_t aclAxis = CalcAxis(descriptor); in ClConcatWorkloadValidate() local
94 size_t aclAxis = CalcAxis(descriptor.m_Parameters); in ClConcatWorkload() local
|
| H A D | ClSplitterWorkload.cpp | 51 unsigned int aclAxis = CalcAclAxis(input.GetNumDimensions(), splitAxis); in ClSplitterWorkloadValidate() local
103 …unsigned int aclAxis = CalcAclAxis(descriptor.m_Parameters.GetNumDimensions(), *splitAxis.begin()); in ClSplitterWorkload() local
|
| H A D | ClGatherNdWorkload.cpp | 73 auto aclAxis = ComputeAclAxis(0, params_K_C_Info); in ClGatherNdWorkloadValidate() local
190 auto aclAxis = ComputeAclAxis(0, paramsInfo); in ClGatherNdWorkload() local
|
| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonLogSoftmaxWorkload.cpp | 26 int aclAxis = ComputeAclAxis(descriptor.m_Axis, input); in NeonLogSoftmaxWorkloadValidate() local
50 int aclAxis = ComputeAclAxis(m_Data.m_Parameters.m_Axis, info.m_InputTensorInfos[0]); in NeonLogSoftmaxWorkload() local
|
| H A D | NeonSoftmaxWorkload.cpp | 26 int aclAxis = ComputeAclAxis(descriptor.m_Axis, input); in NeonSoftmaxWorkloadValidate() local
50 int aclAxis = ComputeAclAxis(m_Data.m_Parameters.m_Axis, info.m_InputTensorInfos[0]); in NeonSoftmaxWorkload() local
|
| H A D | NeonGatherWorkload.cpp | 22 int aclAxis = ComputeAclAxis(descriptor.m_Axis, input); in NeonGatherWorkloadValidate() local
43 int aclAxis = ComputeAclAxis(descriptor.m_Parameters.m_Axis, info.m_InputTensorInfos[0]); in NeonGatherWorkload() local
|
| H A D | NeonArgMinMaxWorkload.cpp | 40 int aclAxis = armnn::numeric_cast<int>(CalcAclAxis(numDims, unsignedAxis)); in NeonArgMinMaxWorkloadValidate() local
70 int aclAxis = armnn::numeric_cast<int>(CalcAclAxis(numDims, unsignedAxis)); in NeonArgMinMaxWorkload() local
|
| H A D | NeonConcatWorkload.cpp | 45 size_t aclAxis = CalcAxis(descriptor); in NeonConcatWorkloadValidate() local
91 size_t aclAxis = CalcAxis(descriptor.m_Parameters); in NeonConcatWorkload() local
|
| H A D | NeonStackWorkload.cpp | 45 int aclAxis = CalcAxis(descriptor.m_Axis, descriptor.m_InputShape.GetNumDimensions()); in NeonStackWorkloadValidate() local
68 …int aclAxis = CalcAxis(descriptor.m_Parameters.m_Axis, descriptor.m_Parameters.m_InputShape.GetNum… in NeonStackWorkload() local
|
| H A D | NeonSplitterWorkload.cpp | 52 unsigned int aclAxis = CalcAclAxis(input.GetNumDimensions(), splitAxis); in NeonSplitterWorkloadValidate() local
103 …unsigned int aclAxis = CalcAclAxis(descriptor.m_Parameters.GetNumDimensions(), *splitAxis.begin()); in NeonSplitterWorkload() local
|
| H A D | NeonGatherNdWorkload.cpp | 71 auto aclAxis = ComputeAclAxis(0, params_K_C_Info); in NeonGatherNdWorkloadValidate() local
|
| /aosp_15_r20/external/armnn/src/backends/aclCommon/ |
| H A D | ArmComputeUtils.hpp | 238 auto aclAxis = (static_cast<T>(dim) - 1); in ComputeSoftmaxAclAxis() local
273 int aclAxis = sign * rank - 1 - armnnAxis; in ComputeAclAxis() local
|
| H A D | ArmComputeTensorUtils.cpp | 400 auto aclAxis = rank - 1 - reversedPerm[i]; in BuildArmComputeTransposeVector() local
|