aboutsummaryrefslogtreecommitdiff
path: root/src/backends/aclCommon/ArmComputeUtils.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/aclCommon/ArmComputeUtils.hpp')
-rw-r--r--src/backends/aclCommon/ArmComputeUtils.hpp13
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();