1 files changed, 3 insertions, 5 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/layers/GenericFullyConnectedLayer.cc b/runtimes/pure_arm_compute/src/internal/layers/GenericFullyConnectedLayer.cc
index 311284efc..28789a801 100644
--- a/runtimes/pure_arm_compute/src/internal/layers/GenericFullyConnectedLayer.cc
+++ b/runtimes/pure_arm_compute/src/internal/layers/GenericFullyConnectedLayer.cc
@@ -17,8 +17,6 @@
 #include "GenericFullyConnectedLayer.h"
 #include "internal/arm_compute.h"
 
-#include <arm_compute/core/Helpers.h>
-
 void GenericFullyConnectedLayer::configure(::arm_compute::ITensor *input,
                                            ::arm_compute::ITensor *weights,
                                            ::arm_compute::ITensor *biases,
@@ -56,9 +54,9 @@ void GenericFullyConnectedLayer::configure(::arm_compute::ITensor *input,
     {
       // reshape
       auto_init_if_empty(*_neon_buffer.info(), _input->info()->clone()->set_tensor_shape(reshape));
-      _generic_reshape.configure(CAST_NE(_input), &_neon_buffer);
+      _generic_reshape.configure(_input, &_neon_buffer);
 
-      _neon_fc.configure(&_neon_buffer, CAST_NE(_weights), CAST_NE(_biases), CAST_NE(_output));
+      _neon_fc.configure(&_neon_buffer, _weights, _biases, _output);
 
       // NOTE _neon_buffer is inaccessible from outside, and thus it is safe to invoke allocate
       // here.
@@ -66,7 +64,7 @@ void GenericFullyConnectedLayer::configure(::arm_compute::ITensor *input,
     }
     else
     {
-      _neon_fc.configure(CAST_NE(_input), CAST_NE(_weights), CAST_NE(_biases), CAST_NE(_output));
+      _neon_fc.configure(_input, _weights, _biases, _output);
     }
   }
 }