1 files changed, 4 insertions, 0 deletions
diff --git a/tests/SimpleTensor.h b/tests/SimpleTensor.h
index ea955fa730..0f79a3899a 100644
--- a/tests/SimpleTensor.h
+++ b/tests/SimpleTensor.h
@@ -44,6 +44,8 @@ namespace arm_compute
 {
 namespace test
 {
+class RawTensor;
+
 /** Simple tensor object that stores elements in a consecutive chunk of memory.
  *
  * It can be created by either loading an image from a file which also
@@ -94,6 +96,8 @@ public:
     using value_type = T;
     using Buffer     = std::unique_ptr<value_type[]>;
 
+    friend class RawTensor;
+
     /** Return value at @p offset in the buffer.
      *
      * @param[in] offset Offset within the buffer.