diff options
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. |