From 3537c2ca7ebf31c1673b9ec2bb0c17b0406bbae0 Mon Sep 17 00:00:00 2001
From: surmeh01 <surabhi.mehta@arm.com>
Date: Fri, 18 May 2018 16:31:43 +0100
Subject: Release 18.05

---
 src/armnn/backends/ArmComputeTensorUtils.hpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'src/armnn/backends/ArmComputeTensorUtils.hpp')

diff --git a/src/armnn/backends/ArmComputeTensorUtils.hpp b/src/armnn/backends/ArmComputeTensorUtils.hpp
index 9a13caf495..84547f9c80 100644
--- a/src/armnn/backends/ArmComputeTensorUtils.hpp
+++ b/src/armnn/backends/ArmComputeTensorUtils.hpp
@@ -9,6 +9,7 @@
 
 #include <arm_compute/core/ITensor.h>
 #include <arm_compute/core/TensorInfo.h>
+#include <arm_compute/core/Types.h>
 
 #include <boost/cast.hpp>
 
@@ -38,6 +39,19 @@ arm_compute::NormalizationLayerInfo BuildArmComputeNormalizationLayerInfo(const
 /// Utility function used to setup an arm_compute::PermutationVector object from an armnn::PermutationVector
 arm_compute::PermutationVector BuildArmComputePermutationVector(const armnn::PermutationVector& vector);
 
+/// Utility function used to setup an arm_compute::PadStrideInfo object from an armnn layer descriptor
+template <typename Descriptor>
+arm_compute::PadStrideInfo BuildArmComputePadStrideInfo(const Descriptor &descriptor)
+{
+    return arm_compute::PadStrideInfo(descriptor.m_StrideX,
+                                      descriptor.m_StrideY,
+                                      descriptor.m_PadLeft,
+                                      descriptor.m_PadRight,
+                                      descriptor.m_PadTop,
+                                      descriptor.m_PadBottom,
+                                      arm_compute::DimensionRoundingType::FLOOR);
+}
+
 /// Sets up the given ArmCompute tensor's dimensions based on the given ArmNN tensor.
 template <typename Tensor>
 void BuildArmComputeTensor(Tensor& tensor, const armnn::TensorInfo& tensorInfo)
-- 
cgit v1.2.1