diff options
Diffstat (limited to 'tests/CL/CLAccessor.h')
-rw-r--r-- | tests/CL/CLAccessor.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/CL/CLAccessor.h b/tests/CL/CLAccessor.h index b1d7a078b1..2f955653c8 100644 --- a/tests/CL/CLAccessor.h +++ b/tests/CL/CLAccessor.h @@ -63,6 +63,8 @@ public: int fixed_point_position() const override; const void *operator()(const Coordinates &coord) const override; void *operator()(const Coordinates &coord) override; + const void *data() const; + void *data(); private: CLTensor &_tensor; @@ -124,6 +126,16 @@ inline int CLAccessor::fixed_point_position() const return _tensor.info()->fixed_point_position(); } +inline const void *CLAccessor::data() const +{ + return _tensor.buffer(); +} + +inline void *CLAccessor::data() +{ + return _tensor.buffer(); +} + inline const void *CLAccessor::operator()(const Coordinates &coord) const { return _tensor.ptr_to_element(coord); |