diff options
Diffstat (limited to 'src/backends/aclCommon/ArmComputeUtils.hpp')
-rw-r--r-- | src/backends/aclCommon/ArmComputeUtils.hpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/backends/aclCommon/ArmComputeUtils.hpp b/src/backends/aclCommon/ArmComputeUtils.hpp index 5b8f983ecc..0f56160051 100644 --- a/src/backends/aclCommon/ArmComputeUtils.hpp +++ b/src/backends/aclCommon/ArmComputeUtils.hpp @@ -122,6 +122,19 @@ ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo(const FullyConnected return fc_info; } +inline arm_compute::InterpolationPolicy ConvertResizeMethodToAclInterpolationPolicy(ResizeMethod resizeMethod) +{ + switch (resizeMethod) + { + case ResizeMethod::Bilinear: + return arm_compute::InterpolationPolicy::BILINEAR; + case ResizeMethod::NearestNeighbor: + return arm_compute::InterpolationPolicy::NEAREST_NEIGHBOR; + default: + throw InvalidArgumentException("Unsupported resize method"); + } +} + inline unsigned int ComputeSoftmaxAclAxis(const armnn::TensorInfo& tensor) { unsigned int dim = tensor.GetNumDimensions(); |