From cde1e8adeacea5c33a1682ef7b05a0ef643463b8 Mon Sep 17 00:00:00 2001 From: Moritz Pflanzer Date: Fri, 8 Sep 2017 09:53:14 +0100 Subject: COMPMID-415: Add tests for ConvolutionLayer reshaped weights Change-Id: I6c1209a2afafccba2cbdbcda16aceb3ae0cc7b4b Reviewed-on: http://mpd-gerrit.cambridge.arm.com/87000 Tested-by: Kaizen Reviewed-by: Gian Marco Iodice --- tests/SimpleTensor.h | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'tests/SimpleTensor.h') 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; + friend class RawTensor; + /** Return value at @p offset in the buffer. * * @param[in] offset Offset within the buffer. -- cgit v1.2.1