diff options
author | Moritz Pflanzer <moritz.pflanzer@arm.com> | 2017-09-08 09:53:14 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | cde1e8adeacea5c33a1682ef7b05a0ef643463b8 (patch) | |
tree | 47e58abdf5bb6ef39db362a2ac777c93b3f76666 /tests/SimpleTensor.h | |
parent | 86b53339679e12c952a24a8845a5409ac3d52de6 (diff) | |
download | ComputeLibrary-cde1e8adeacea5c33a1682ef7b05a0ef643463b8.tar.gz |
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>
Diffstat (limited to 'tests/SimpleTensor.h')
-rw-r--r-- | tests/SimpleTensor.h | 4 |
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. |