COMPMID-415: Add tests for ConvolutionLayer reshaped weights
Change-Id: I6c1209a2afafccba2cbdbcda16aceb3ae0cc7b4b
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/87000
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
diff --git a/tests/SimpleTensor.h b/tests/SimpleTensor.h
index ea955fa..0f79a38 100644
--- a/tests/SimpleTensor.h
+++ b/tests/SimpleTensor.h
@@ -44,6 +44,8 @@
{
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 @@
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.