commit a437638028c216a03572593b9d4e3532df2a308d
author Anthony Barbier <anthony.barbier@arm.com> Wed Apr 12 15:12:46 2017 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Wed Apr 12 17:42:01 2017 +0100
tree b2aeef6797fdc620419431b4b3d3c4f46febf69e
parent 68a98dc29106a4c8c34e8cac542fe0bdad4ad531

arm_compute v17.04

src/runtime/NEON/functions/NEHOGGradient.cpp

diff --git a/src/runtime/NEON/functions/NEHOGGradient.cpp b/src/runtime/NEON/functions/NEHOGGradient.cpp
index c82f0af..c5b37f4 100644
--- a/src/runtime/NEON/functions/NEHOGGradient.cpp
+++ b/src/runtime/NEON/functions/NEHOGGradient.cpp
@@ -45,11 +45,8 @@
 
     // Allocate image memory
     TensorInfo info(shape_img, Format::S16);
-    info.auto_padding();
     _gx.allocator()->init(info);
-    _gx.allocator()->allocate();
     _gy.allocator()->init(info);
-    _gy.allocator()->allocate();
 
     // Initialise derivate kernel
     _derivative.configure(input, &_gx, &_gy, border_mode, constant_border_value);
@@ -67,6 +64,10 @@
         k->configure(&_gx, &_gy, output_magnitude, output_phase);
         _mag_phase = std::move(k);
     }
+
+    // Allocate intermediate tensors
+    _gx.allocator()->allocate();
+    _gy.allocator()->allocate();
 }
 
 void NEHOGGradient::run()