commit 52ba29e936b8e711e8acdfe819e36f884d4f3fe1
author Jenkins <bsgcomp@arm.com> Wed Aug 29 15:32:11 2018 +0000
committer Michele Di Giorgio <michele.digiorgio@arm.com> Thu Aug 30 17:15:09 2018 +0100
tree 23a8e15208ad7fcf07a37ed9119979cde01dcdc8
parent e2542c9f35ca427286822cd0c9296f49914f78b0

arm_compute v18.08

src/runtime/ITensorAllocator.cpp

diff --git a/src/runtime/ITensorAllocator.cpp b/src/runtime/ITensorAllocator.cpp
index 8294201..087f324 100644
--- a/src/runtime/ITensorAllocator.cpp
+++ b/src/runtime/ITensorAllocator.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2016, 2017 ARM Limited.
+ * Copyright (c) 2016-2018 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -31,13 +31,14 @@
 using namespace arm_compute;
 
 ITensorAllocator::ITensorAllocator()
-    : _info()
+    : _info(), _alignment(0)
 {
 }
 
-void ITensorAllocator::init(const TensorInfo &input)
+void ITensorAllocator::init(const TensorInfo &input, size_t alignment)
 {
-    _info = input;
+    _info      = input;
+    _alignment = alignment;
 }
 
 TensorInfo &ITensorAllocator::info()
@@ -49,3 +50,8 @@
 {
     return _info;
 }
+
+size_t ITensorAllocator::alignment() const
+{
+    return _alignment;
+}