diff options
Diffstat (limited to 'src/backends/aclCommon/ArmComputeUtils.hpp')
| -rw-r--r-- | src/backends/aclCommon/ArmComputeUtils.hpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/backends/aclCommon/ArmComputeUtils.hpp b/src/backends/aclCommon/ArmComputeUtils.hpp index 3d5b9cae01..b4673f7b31 100644 --- a/src/backends/aclCommon/ArmComputeUtils.hpp +++ b/src/backends/aclCommon/ArmComputeUtils.hpp @@ -120,4 +120,14 @@ ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo(const FullyConnected return fc_info; } +inline unsigned int ComputeSoftmaxAclAxis(const armnn::TensorInfo& tensor) +{ + unsigned int dim = tensor.GetNumDimensions(); + + BOOST_ASSERT(dim != 0); + + // Currently ArmNN support axis 1. + return dim - 1; +} + } // namespace armnn |